__uint2bfloat16_rn

Applicability

Product	Supported
Atlas 350 Accelerator Card	√
Atlas A3 training product/Atlas A3 inference product	x
Atlas A2 training product/Atlas A2 inference product	x
Atlas 200I/500 A2 inference product	x
Atlas inference product AI Core	x
Atlas inference product Vector Core	x
Atlas training product	x

Function Usage

Converts uint32 data to the bfloat16 type in line with the CAST_RINT mode and returns the converted value.

Prototype

inline bfloat16_t __uint2bfloat16_rn(const unsigned int x)

Parameters

**Table 1** Parameters
Parameter	Input/Output	Description
x	Input	Source operand.

Returns

The bfloat16 data converted from the input in line with the CAST_RINT mode.

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.

#include "simt_api/asc_bf16.h"

Examples

For SIMD and SIMT programming:

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

Parent topic: Type Conversion