__ushort_as_bfloat16

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

Reinterprets the bits of an unsigned short int as a bfloat16 value, reading the stored bits in bfloat16 format.

Prototype

inline bfloat16_t __ushort_as_bfloat16(const unsigned short int x)

Parameters

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

Returns

bfloat16 value reinterpreted from the bits of the unsigned short int

Restrictions

For SIMT programming, this API is not supported.

Header Files 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__ushort_as_bfloat16(__gm__ bfloat16_t* dst, __gm__ unsigned short int* x)
{
    int idx = threadIdx.x + blockIdx.x * blockDim.x;
    dst[idx] = __ushort_as_bfloat16(x[idx]);
}

Parent topic: Type Conversion