在自动混合精度策略(AMP)场景下判断设备是否支持 float16。
device (str|None,可选) - 指定要查询的设备,它可以是 cpu、 gpu、 xpu、gpu:x、xpu:x。其中,x 是 GPU、 XPU 的编号。如果 device 为 None, 则查询当前设备(与飞桨安装版本保持一致), 默认为 None。
device
>>> import paddle >>> paddle.amp.is_float16_supported() # True or False False