| Add ReduceProdV2 operator implementation (issue #249)
Co-authored-by: hitLeechi<2084486416@qq.com>
# message auto-generated for no-merge-commit merge:
!401 merge feat/op1 into master
Add ReduceProdV2 operator implementation (issue #249)
Created-by: hitLeechi
Commit-by: hitLeechi
Merged-by: cann-robot
Description: ## 描述
使用AscendC对TBE实现的ReduceSum算子进行重构,本质上是reduceSumV2算子的对指变换重构实现了AscendC实现的算子对Atlas 200/500 A2推理产品和Atlas 800I/T A2硬件的适配。
## 关联的Issue
关联Issue:https://gitcode.com/cann/ops-math/issues/249
## 测试
参见评论附件
## 文档更新
## 局限性
最高支持二维输入张量,仅支持非负数输入,数据类型仅支持float16,32,DumpTensor严重影响性能,但是不添加DumpTensor指令Exp接口似乎没有执行,后续将完成任意维度支持负值的版本
## 类型标签
<!-- [ ] 表示选中 -->
- [x] Bug修复
- [x] 新特性
- [ ] 性能优化
- [ ] 文档更新
- [ ] 其他,请描述:
See merge request: cann/ops-math!401 | 2 个月前 |