is_bf16_supported

paddle.cuda. is_bf16_supported ( including_emulation: bool = True ) → bool [source]

Return a bool indicating if the current CUDA/ROCm device supports dtype bfloat16.

Parameters: including_emulation (bool = True) – Whether to treat software-emulated BF16 as supported; if False, only native hardware BF16 support is considered.
Returns: A boolean value which indicates whether the current CUDA/ROCm device supports dtype bfloat16.
Return type: bool

Examples

           >>> import paddle

>>> paddle.device.is_bf16_supported()
>>> # paddle.cuda.is_bf16_supported() is an alias of paddle.device.is_bf16_supported()
>>> paddle.cuda.is_bf16_supported()