下载
中文
注册
我要评分
文档获取效率
文档正确性
内容完整性
文档易理解
在线提单
论坛求助
昇腾小AI

ToFloat

功能说明

bfloat16_t类型标量数据转换成float类型标量数据。

函数原型

1
__aicore__ inline float ToFloat(const bfloat16_t& bVal)

参数说明

表1 接口参数说明

参数名称

输入/输出

含义

bVal

输入

bfloat16_t类型标量数据

支持的型号

Atlas A2训练系列产品/Atlas 800I A2推理产品

注意事项

返回值

转换后的float类型标量数据。

调用示例

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
float m = 3.0f;
int dataLen = 32;
bfloat16_t n = AscendC::ToBfloat16(m);
AscendC::TPipe pipe;
AscendC::TQue<AscendC::QuePosition::VECIN, 1> inQueueSrcVecIn;
pipe.InitBuffer(inQueueSrcVecIn, 1, dataLen * sizeof(bfloat16_t));
AscendC::LocalTensor<bfloat16_t> dstLocal = inQueueSrcVecIn.AllocTensor<bfloat16_t>();
// 输入bfloat16_t类型,做加法运算后,输出bfloat16_t类型tensor
CalcFunc(n, dstLocal, dataLen); // 需实现     
{ 
    AscendC::TPipe pipe;         
    AscendC::TQue<AscendC::QuePosition::VECIN, 1> inQueueSrcVecIn;         
    pipe.InitBuffer(inQueueSrcVecIn, 1, dataLen * sizeof(float));
    AscendC::LocalTensor<float> srcLocal = inQueueSrcVecIn.AllocTensor<float>();
    AscendC::Duplicate(srcLocal, float(4.0f), dataLen);
    float t = AscendC::ToFloat(n);// 对标量进行加法,不支持bfloat16_t,需要先转换成float
    Adds(srcLocal, srcLocal, t, dataLen);
    Cast(dstLocal, srcLocal, AscendC::RoundMode::CAST_ROUND, dataLen);
}
搜索结果
找到“0”个结果

当前产品无相关内容

未找到相关内容,请尝试其他搜索词