o Õj› ã@sŒddlmZmZmZmZddlmZmZddlmZm Z m Z ddl m Z m Z m Z ddlmZddlmZmZgd¢Zdd d „Zdd d „Zd S)é)Ú auto_castÚdecorateÚ amp_guardÚ amp_decorate)Ú white_listÚ black_list)Ú debuggingÚ grad_scalerÚaccuracy_compare)Ú GradScalerÚ AmpScalerÚOptimizerStateé)Úcore)Ú_current_expected_placeÚ_get_paddle_place)rr rÚis_float16_supportedÚis_bfloat16_supportedNcCó |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©Zdevice©rúT/var/www/html/Deteccion_Ine/venv/lib/python3.10/site-packages/paddle/amp/__init__.pyr5ó ÿý 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)rrrrrrrrrPrr)N)rrrrZ amp_listsrrÚrr r r r r Z paddle.baserZpaddle.base.frameworkrrÚ__all__rrrrrrÚs