__float2bfloat16_rn

Applicability

Function Usage

Converts a floating-point number to bfloat16 precision in line with the CAST_RINT mode and returns the converted value.

Prototype

1

inline bfloat16_t __float2bfloat16_rn(const float x)

Parameters

Returns

The bfloat16 data converted from the input in line with the CAST_RINT mode. Special scenarios are described as follows:

• If x is nan, the return value is nan.
• If x is inf, the return value is inf.
• If x is -inf, the return value is -inf.

Restrictions

For SIMT programming, this API is not supported.

Header File to Be Included

To use this API, the simt_api/asc_bf16.h header file must be included.

1

#include "simt_api/asc_bf16.h"

Examples

1
2
3
4
5

__simt_vf__ __launch_bounds__(1024) inline void kernel__float2bfloat16_rn(__gm__ bfloat16_t* dst, __gm__ float* x)
{
    int idx = threadIdx.x + blockIdx.x * blockDim.x;
    dst[idx] = __float2bfloat16_rn(x[idx]);
}