CANN asc-devkit IsFinite接口文档

发布时间:2026/7/17 18:35:33

CANN asc-devkit IsFinite接口文档 IsFinite【免费下载链接】asc-devkit本项目是CANN 推出的昇腾AI处理器专用的算子程序开发语言原生支持C和C标准规范主要由类库和语言扩展层构成提供多层级API满足多维场景算子开发诉求。项目地址: https://gitcode.com/cann/asc-devkit产品支持情况Ascend 950PR/Ascend 950DT支持Atlas A3 训练系列产品/Atlas A3 推理系列产品不支持Atlas A2 训练系列产品/Atlas A2 推理系列产品不支持Atlas 200I/500 A2 推理产品不支持Atlas 推理系列产品AI Core不支持Atlas 推理系列产品Vector Core不支持Atlas 训练系列产品不支持功能说明按元素判断输入的浮点数是否非NAN、非±INF输出结果为浮点数或者布尔值。对于非NAN或者非±INF的输入数据当输出为浮点类型时对应位置的结果为该浮点类型的1反之为0当输出为bool类型时对应位置的结果为true反之为false。计算公式如下当输出为浮点类型时当输出为bool类型时函数原型templatetypename T, typename U __aicore__ inline void IsFinite(const LocalTensorU dst, const LocalTensorT src, uint32_t calCount)参数说明表1模板参数说明参数名描述T源操作数的数据类型。支持的数据类型为half、bfloat16_t、float。U目的操作数的数据类型。支持的数据类型为bool、half、bfloat16_t、float。表2接口参数说明参数名称输入/输出含义dst输出目的操作数。类型为LocalTensor支持的TPosition为VECIN/VECCALC/VECOUT。目的操作数的数据类型和源操作数保持一致或者目的操作数的数据类型为bool类型。当前支持的数据类型组合请见表3。src输入源操作数。类型为LocalTensor支持的TPosition为VECIN/VECCALC/VECOUT。calCount输入参与计算的元素个数。表3输入输出支持的数据类型组合srcDtypedstDtypehalfhalfhalfboolfloatfloatfloatboolbfloat16_tbfloat16_tbfloat16_tbool返回值说明无约束说明不支持源操作数与目的操作数地址重叠。操作数地址偏移对齐要求请参见通用说明和约束。调用示例完整算子样例请参考isfinite算子样例。// dstLocal: 存放IsFinite计算结果的Tensor // srcLocal存放IsFinite计算的输入Tensor // 需要参与计算的元素个数为8 AscendC::IsFinite(dstLocal, srcLocal, 8);结果示例如下输入的数据类型为float输出的数据类型为bool 输入数据(src): [1.0,inf,3.0,4.0,nan,6.0,-inf,8.0] 输出数据(dst): [true,false,true,true,false,true,false,true]【免费下载链接】asc-devkit本项目是CANN 推出的昇腾AI处理器专用的算子程序开发语言原生支持C和C标准规范主要由类库和语言扩展层构成提供多层级API满足多维场景算子开发诉求。项目地址: https://gitcode.com/cann/asc-devkit创作声明:本文部分内容由AI辅助生成(AIGC),仅供参考

相关新闻