已关闭
[Requirement|需求建议]: 修改abs算子kernel代码,使其精度与竞品保持一致 #1336
gaomingyuan创建于 4月22日关闭于 4月23日
sunchun
4月22日 评论:
4月22日 评论:
/assign @qq_52874943


4月22日 将 qq_52874943 设为负责人
4月23日 修改了issue 的描述
gaomingyuan
4月23日 评论:
4月23日 评论:
/assign


CANN-robot
4月23日 评论:
4月23日 评论:
4月23日 关闭了 issue
4月23日 添加了label:resolved
Thanks for sending an requirement! Please fill in the following template to help quickly solve your problem.
一、背景信息 (必填)
为充分发挥Ascend950的算力,abs需要在该平台修改kernel,使其精度与竞品保持一致
二、价值/作用 (必填)
abs 算子新增 complex64 和 complex32 输入类型,是保障复数域机器学习计算稳定性的关键改进,通过算子层面对复数模长进行标准化、高精度计算,有效避免了手动实现带来的精度损失与数值溢出风险,广泛应用于通信信号分析、复值神经网络损失计算和量子机器学习的概率建模中。
三、设计方案 (必填)
通过提取a的指数因子方式,对标标杆的实现过程,实现精度一致。
3.1 使能方式(涉及哪些框架:如Aclnn直调、Pytorch训练等)
3.2 总体设计
3.2.1 算子支持的数据类型
3.2.2 host侧设计
3.2.3 kernel侧设计
3.3 支持硬件
3.4 算子约束限制
💡 备注(选填)