| 文件 | 最后提交记录 | 最后更新时间 |
|---|---|---|
个人-AscendC实现ReduceSum算子贡献 Co-authored-by: xiaopei-1<1550747408@qq.com> # message auto-generated for no-merge-commit merge: !368 merge reducesum into master 个人-AscendC实现ReduceSum算子贡献 Created-by: xiaopei-1 Commit-by: xiaopei-1 Merged-by: cann-robot Description: ## 描述 使用AscendC对TBE实现的ReduceSum算子进行重构,实现了AscendC实现的算子对Atlas 200/500 A2推理产品和Atlas 800I/T A2硬件的适配。 ## 关联的Issue 关联Issue:https://gitcode.com/cann/ops-math/issues/219 ## 测试 测试用例  测试结果  ## 文档更新 ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [x ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他,请描述: See merge request: cann/ops-math!368 | 7 个月前 | |
refactor: 移除多个算子 tiling 文件中冗余的 tiling_util.h 头文件引用 Co-authored-by: songkai111<songkai16@huawei.com> # message auto-generated for no-merge-commit merge: !2962 merge master into master refactor: 移除多个算子 tiling 文件中冗余的 tiling_util.h 头文件引用 Created-by: songkai111 Commit-by: songkai111 Merged-by: cann-robot Description: ## 描述 本 PR 对多个算子的 tiling 代码进行头文件引用清理和代码格式化,移除不再需要的 op_host/tiling_util.h 引用,统一代码风格。 ### 改动原因 - 多个算子的 tiling 文件中仍残留 #include "op_host/tiling_util.h" 头文件引用,该头文件已不再被使用,属于冗余依赖 - 部分文件的代码格式不统一(行宽、缩进等),需要进行格式化对齐 ### 改动方法 1. **移除冗余头文件**:在以下 24 个算子的 tiling 文件中移除 #include "op_host/tiling_util.h": - axpy_v2, bitwise_and, diag_part, equal, erf, floor, nan_to_num, pow, pow2, real, real_div, reciprocal, reduce_max_v2, reduce_min_v2, reduce_sum_v2, round, rsqrt, rsqrt_grad, select_v2, sign, sign_bits_pack, sign_bits_unpack, sqrt, tanh_grad 2. **equal 算子**:额外移除 #include "op_host/tiling_templates_registry.h",替换为 register/op_impl_registry.h、tiling/platform/platform_ascendc.h、graph/utils/type_utils.h 3. **real 算子**:在移除冗余头文件的基础上,进行代码格式化调整,统一行宽和缩进风格 4. **CI 脚本**(scripts/ci/gen_ci_cmd.py):将 experimental 算子的 CI example 命令生成逻辑从"使用默认算子"改为直接跳过,避免生成无效的测试命令 ## 关联的Issue - #1666 ## 测试 编译验证通过,移除未使用的头文件不影响功能。 ## 文档更新 无 ## 类型标签 - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [x] 其他,请描述:代码重构 - 移除冗余头文件引用和代码格式化 See merge request: cann/ops-math!2962 | 2 个月前 | |
个人-AscendC实现ReduceSum算子贡献 Co-authored-by: xiaopei-1<1550747408@qq.com> # message auto-generated for no-merge-commit merge: !368 merge reducesum into master 个人-AscendC实现ReduceSum算子贡献 Created-by: xiaopei-1 Commit-by: xiaopei-1 Merged-by: cann-robot Description: ## 描述 使用AscendC对TBE实现的ReduceSum算子进行重构,实现了AscendC实现的算子对Atlas 200/500 A2推理产品和Atlas 800I/T A2硬件的适配。 ## 关联的Issue 关联Issue:https://gitcode.com/cann/ops-math/issues/219 ## 测试 测试用例  测试结果  ## 文档更新 ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [x ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他,请描述: See merge request: cann/ops-math!368 | 7 个月前 | |
支持下一代cos_stateless_bernoulli_stateless_random_normal_v2算子实现 Co-authored-by: z00940181<zhanghaodong30@huawei.com> # message auto-generated for no-merge-commit merge: !548 merge master into master 支持下一代cos_stateless_bernoulli_stateless_random_normal_v2算子实现 Created-by: hawdonz Commit-by: z00940181 Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动,包括改动的原因和所采取的方法。--> ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue,请在这里提供Issue链接。--> <!-- 如果这个PR是为了解决特定的问题单,请在这里描述问题单单号。--> ## 测试 <!--描述进行了哪些测试来验证你的改动。包括但不限于二级冒烟、算子泛化等。--> ## 文档更新 <!--如果这个PR包含文档的更新,请在这里指出。例如:更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他,请描述: See merge request: cann/ops-math!548 | 7 个月前 | |
个人-AscendC实现ReduceSum算子贡献 Co-authored-by: xiaopei-1<1550747408@qq.com> # message auto-generated for no-merge-commit merge: !368 merge reducesum into master 个人-AscendC实现ReduceSum算子贡献 Created-by: xiaopei-1 Commit-by: xiaopei-1 Merged-by: cann-robot Description: ## 描述 使用AscendC对TBE实现的ReduceSum算子进行重构,实现了AscendC实现的算子对Atlas 200/500 A2推理产品和Atlas 800I/T A2硬件的适配。 ## 关联的Issue 关联Issue:https://gitcode.com/cann/ops-math/issues/219 ## 测试 测试用例  测试结果  ## 文档更新 ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [x ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他,请描述: See merge request: cann/ops-math!368 | 7 个月前 | |
doc tools 工具扫描问题修改 Co-authored-by: gitee-yanglulu<yanglulul@h-partners.com> # message auto-generated for no-merge-commit merge: !1938 merge master into master doc tools 工具扫描问题修改 Created-by: gitee-yanglulu Commit-by: gitee-yanglulu Merged-by: cann-robot Description: doc tools 工具扫描问题修改 See merge request: cann/ops-math!1938 | 4 个月前 |
ReduceSumV2
产品支持情况
| 产品 | 是否支持 |
|---|---|
| Atlas A2 训练系列产品/Atlas A2 推理系列产品 | √ |
功能说明
- 算子功能:对输入张量x沿指定维度计算给定张量的和。
参数说明
| 参数名 | 输入/输出/属性 | 描述 | 数据类型 | 数据格式 |
|---|---|---|---|---|
| self | 输入 | x | float、half | ND |
| axis | 属性 | axes | int | ND |
| keepdims | 属性 | keepdims | int | ND |
| y | 输出 | y | float、half | ND |
约束说明
无
待更新
目前支持float32、float16数据类型,仅支持2维,axis仅支持1根轴reduce,keepdims支持0/1,后续会持续改进
调用说明
| 调用方式 | 样例代码 | 说明 |
|---|---|---|
| aclnn接口 | test_reduce_sum_v2 | 通过[aclnnReduceSumV2]接口方式调用reduce_sum_v2算子。 |