已合并
feat(logical_or): 支持arch35平台多数据类型kernel直接计算 #2930
hahaha22创建于 5月25日
feat(logical_or): 支持arch35平台多数据类型kernel直接计算 #2930
已合并
Pull Request已成功合入, 合并人@CANN-robot
(感谢 hahaha22 的贡献)此处折叠了79条消息 查看更多
5月29日 添加了label:lgtm
5月29日 关闭了关联的issue
5月29日 合入了pull request


描述
logical_or算子在arch35(Ascend 950PR/Ascend 950DT)平台上新增INT8、UINT8、INT16、INT32、INT64、FLOAT16、BFLOAT16、FLOAT数据类型的kernel直接计算支持。此前仅BOOL类型可走AiCore kernel,其他类型需先Cast到BOOL再计算。在RegBase平台上,当两个输入类型相同且属于kernel支持的数据类型时,可直接调用kernel计算,减少Cast开销。
改动原因
改动方法
LogicalOrCompute<uint8_t>:BOOL类型专用,直接位OR后输出LogicalOrIntegralCompute<T>:整数类型专用,Or→Compare(NE 0)→Select(1/0)→输出BOOLLogicalOrFloatCompute<T, IntT>:浮点类型专用,使用自定义MicroAPI OpLogicalOrFloatCustom实现位OR+Compare+Select,输出BOOL关联的Issue
测试
文档更新
类型标签