is_bfloat16_supported¶

在自动混合精度策略（AMP）场景下判断设备是否支持 bfloat16。

参数¶

device (str|None，可选) - 指定要查询的设备，它可以是 cpu、 gpu、 xpu、gpu:x、xpu:x。其中，x 是 GPU、 XPU 的编号。如果 device 为 None，则查询当前设备（与飞桨安装版本保持一致），默认为 None。

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

         