is_float16_supported

paddle.amp. is_float16_supported ( device: Optional[str] = None ) → bool [source]

Determine whether the place supports float16 in the auto-mixed-precision training.

Parameters: 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

           >>> import paddle
>>> paddle.amp.is_float16_supported() # True or False
False

          