__int2bfloat16_rn
产品支持情况
产品 |
是否支持 |
|---|---|
Atlas 350 加速卡 |
√ |
x |
|
x |
|
x |
|
x |
|
x |
|
x |
功能说明
将int32类型数据转换为bfloat16类型数据，并遵循CAST_RINT模式，返回转换后的值。
函数原型
1 | inline bfloat16_t __int2bfloat16_rn(const int x) |
参数说明
参数名 |
输入/输出 |
描述 |
|---|---|---|
x |
输入 |
源操作数。 |
返回值说明
输入的uint32数据转换成的bfloat16数据，并四舍五入到最接近的偶数。
约束说明
SIMT编程场景当前不支持使用该接口。
需要包含的头文件
使用该接口需要包含"simt_api/asc_bf16.h"头文件。
1 | #include "simt_api/asc_bf16.h" |
调用示例
SIMD与SIMT混合编程场景：
1 2 3 4 5 | __simt_vf__ __launch_bounds__(1024) inline void kernel__int2bfloat16_rn(__gm__ bfloat16_t* dst, __gm__ int32_t* x) { int idx = threadIdx.x + blockIdx.x * blockDim.x; dst[idx] = __int2bfloat16_rn(x[idx]); } |
父主题： 类型转换