| Files | Last commit | Last update |
|---|---|---|
master翻译合入 Co-authored-by: caiwenwen<caiwenwen6@h-partners.com> # message auto-generated for no-merge-commit merge: !6751 merge master into master master翻译合入 Created-by: caiwenwen Commit-by: caiwenwen Merged-by: cann-robot Description: ## 描述 合入math仓翻译的内容 ## 关联的Issue https://gitcode.com/cann/ops-math/issues/4170 ## 测试 <!--描述进行了哪些测试来验证你的改动。包括但不限于二级冒烟、算子泛化等。--> ## 文档更新 docs/context，aclnn等商发文档 ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [x] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!6751 | 13 days ago | |
math仓算子错误日志整改 Co-authored-by: qq_52874943<gaomingyuan3@h-partners.com> # message auto-generated for no-merge-commit merge: !5345 merge master into master math仓算子错误日志整改 Created-by: qq_52874943 Commit-by: qq_52874943 Merged-by: cann-robot Description: ## 描述 本 PR 针对 math 仓算子日志进行整改（关联 Issue #2853），主要目的是规范算子错误日志的级别与内容。改动覆盖约 35 个文件，核心内容包括三类：将 examples/test_geir_*.cpp 示例程序中 GE 初始化失败、加图失败、运行图失败、反初始化失败等错误场景的日志级别由 INFO 修正为 ERROR；修正 op_kernel_aicpu、op_api、tiling、framework 插件中参数校验及错误提示日志的英文语法与拼写；同时清理行尾多余空白字符。整体为纯日志与文档性整改，不涉及算子计算逻辑与对外接口行为变更。 ## 关联的Issue 关联Issue[#2853](https://gitcode.com/cann/ops-math/issues/2853) ## 测试 本地已完成新精度对比与性能测试，通过静态检查、蓝区冒烟、算子ST以及算子二级冒烟。 ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [x] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!5345 | 2 months ago | |
迁移开源TF插件--math Co-authored-by: yaochen<yaochen15@huawei.com> # message auto-generated for no-merge-commit merge: !4308 merge transfer-math into master 迁移开源TF插件--math Created-by: nextyale Commit-by: yaochen Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动，包括改动的原因和所采取的方法。--> ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。--> <!-- 如果这个PR是为了解决特定的问题单，请在这里描述问题单单号。--> ## 测试 <!--描述进行了哪些测试来验证你的改动。包括但不限于二级冒烟、算子泛化等。--> ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [x] 其他，请描述：迁移TF插件 See merge request: cann/ops-math!4308 | 3 months ago | |
fix(add_n): 补齐输入dtype一致性校验并修复空TensorList段错误(issue #3652) Co-authored-by: chenyifeng<chenyifeng27@h-partners.com> # message auto-generated for no-merge-commit merge: !6798 merge fix_addn_1029 into master fix(add_n): 补齐输入dtype一致性校验并修复空TensorList段错误(issue #3652) Created-by: chen_0715 Commit-by: chenyifeng Merged-by: cann-robot Description: ## 描述 修复aclnnAddN算子两个问题（issue #3652）： 1. 补齐输入dtype一致性校验：原实现仅逐项检查每个输入是否在支持列表内、仅比对tensors[0]与out的dtype，且依赖的OP_CHECK_DTYPE_NOT_SAME宏存在逻辑缺陷，导致tensors内部dtype不一致时校验无法正常拦截。现改为在CheckDtypeValid中遍历tensors全部元素，逐一校验其dtype与out一致，不一致返回161002，与文档约定（tensors内所有输入及out数据类型需保持一致）对齐。 2. 修复空TensorList段错误：原CheckParams无条件访问(*tensors)[0]，空TensorList（Size()==0）时发生越界访问导致段错误。现改为空列表提前放行（文档约定支持空TensorList，此时输出为空Tensor，由GetWorkspaceSize的空列表分支返回workspaceSize=0）。 ## 关联的Issue - #3652：aclnnAddN缺失tensors内部dtype一致性校验，空TensorList触发段错误 - [#3558](https://gitcode.com/cann/ops-math/issues/3558)： ## 测试 新增3个UT用例（math/add_n/tests/ut/op_api/test_aclnn_add_n.cpp）： - 空TensorList：返回ACLNN_SUCCESS且workspaceSize为0； - tensors内部dtype不一致（FLOAT+INT32）：返回161002； - out与输入dtype不一致（输入FLOAT、out INT32）：返回161002。 ## 文档更新 无文档更新，本次仅修改实现代码与UT。 ## 类型标签 - [x] Bug修复 See merge request: cann/ops-math!6798 | 11 days ago | |
feat: Infershape整改，25个math算子InferDataType注册迁移至op_graph目录（IMPL_OP独立链）及UT看护 Co-authored-by: ywx1080114<yuanbin22@huawei.com> # message auto-generated for no-merge-commit merge: !6392 merge infershape-dtype into master feat: Infershape整改，25个math算子InferDataType注册迁移至op_graph目录（IMPL_OP独立链）及UT看护 Created-by: yuanbin_22 Commit-by: ywx1080114 Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动，包括改动的原因和所采取的方法。--> 作为 Infershape 整改的一部分，为 25 个 math 算子在 op_graph/ 目录新增 InferDataType（输出数据类型推导）独立注册及 UT 看护，共 50 个文件（+1757 -0，纯新增），对齐 ops-nn 仓方案（参考 PR#9839 样板）： **注册方案（IMPL_OP 独立链）：** - 每个算子新增 op_graph/<op>_graph_infer.cpp，通过 IMPL_OP(X).InferDataType(InferDataTypeX) 独立注册链注册，与 op_host 的 IMPL_OP_INFERSHAPE（InferShape）链解耦； - 涉及 25 个算子：abs_grad、add_n、assign_add、assign_sub、cast、cos、erf、exp、expm1、floor、log1p、logical_not、masked_scale、nan_to_num、neg、reciprocal、rsqrt、rsqrt_grad、sign、sin、sqrt、sqrt_grad、square、tanh、trunc； - 原 35 个算子中的另外 10 个（atan、ceil、is_finite、is_inf、is_nan、is_neg_inf、is_pos_inf、log、right_shift、round）的 op_graph 迁移已单独合入 master，不在本 PR 范围； **实现分类：** - 24 个算子输出 dtype 与第一个输入一致（SetOutputDataType(0, GetInputDataType(0))）； - Cast 按属性推导并**新增枚举范围校验**：从 gert::RuntimeAttrs 读取 dst_type 属性，校验合法值域为 [0, DT_MAX) 且排除 DT_UNDEFINED（=28），非法值在推导处直接报错返回 GRAPH_FAILED（与 arch31 tiling 的 TransAclToGeDataType 白名单校验同款防御，防止非法值静默写入输出 desc）。 **UT 看护（25 个测试文件，26 个用例）：** - 每算子新增 tests/ut/op_graph/test_<op>_graph_infer_pass.cpp，直接调用导出的 ops::InferDataTypeX 函数，通过 OpInferDataTypeContextBuilder 构造上下文，参数化断言推导返回 GRAPH_SUCCESS 且输出 dtype 符合预期； - Cast 参数化覆盖 dst_type 集合（含 FP16/FLOAT/BF16/INT 系列等），并新增非法值专项用例：dst_type 为 -1/DT_UNDEFINED/DT_MAX/999 时断言推导返回 GRAPH_FAILED。 ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。--> <!-- 如果这个PR是为了解决特定的问题单，请在这里描述问题单单号。--> - 关联 Issue #3792 ## 测试 <!--描述进行了哪些测试来验证你的改动。包括但不限于二级冒烟、算子泛化等。--> - 25 个算子各新增 InferDataTypeTest 参数化 UT 用例（直接函数调用），全部通过； - Cast 额外新增非法 dst_type 专项用例（-1/DT_UNDEFINED/DT_MAX/999 断言 GRAPH_FAILED），覆盖枚举校验防御路径。 ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> - 无需文档更新。 ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!6392 | 7 days ago | |
math仓addn算子tilingubsize大小修改 Co-authored-by: qq_52874943<gaomingyuan3@h-partners.com> # message auto-generated for no-merge-commit merge: !6917 merge master into master math仓addn算子tilingubsize大小修改 Created-by: qq_52874943 Commit-by: qq_52874943 Merged-by: cann-robot Description: <!--在这里详细描述你的改动，包括改动的原因和所采取的方法。--> ## 改动原因 math/add_n 算子在 arch35 下的 tiling 中，fullload（TILINGKEY_FULLLOAD）策略的判定使用整个 ub 容量 ubSize_ 作为上界。当设备 ub 空间较大时，会让更多输入进入 fullload 分支，导致单 block 一次性加载的数据量过大，占用过多 ub 空间。为控制 fullload 策略下的数据加载量、避免 ub 占用过高，需要给该判定加上限。 ## 改动方法 在 math/add_n/op_host/arch35/add_n_tiling_arch35.cpp 的 AddNTiling::SetTilingKey() 中： - 新增常量 ADDN_UNIFIED_FULLLOAD_UB_SIZE = 184 * 1024（184KB），作为 fullload 判定使用的 ub 上限； - 将原判定条件中的上界 ubSize_ 替换为 fullLoadUbSize = std::min(ubSize_, ADDN_UNIFIED_FULLLOAD_UB_SIZE)，即 fullload 判定同时受实际 ub 容量与 184KB 上限约束； - 相应引入 <algorithm> 头文件以使用 std::min。 改动后，仅当 (sizeN_ + 1) * sum2 * dtypeSize_ 小于该受限上界时才走 fullload 分支，从而限制 fullload 策略下的数据加载量。 ## 关联的Issue https://atomgit.com/cann/ops-math/issues/4371 ## 测试 <!--描述进行了哪些测试来验证你的改动。包括但不限于二级冒烟、算子泛化等。--> - 二级冒烟：add_n 算子相关用例通过； - 算子泛化：add_n 相关泛化用例通过，功能与精度符合预期； - 受影响文件仅 math/add_n/op_host/arch35/add_n_tiling_arch35.cpp，fullload 分支命中策略变化，未命中 fullload 的分支行为保持不变。 ## 文档更新 本次为 tiling 内部参数调整，不改变算子对外接口与行为说明，暂无文档更新。 ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [x] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!6917 | 10 days ago | |
refactor: arch35 算子 Regbase Reg 层旧接口替换为新接口 Co-authored-by: ywx1080114<yuanbin22@huawei.com> # message auto-generated for no-merge-commit merge: !4527 merge fix/api-rename-regbase into master refactor: arch35 算子 Regbase Reg 层旧接口替换为新接口 Created-by: yuanbin_22 Commit-by: ywx1080114 Merged-by: cann-robot Description: ## 描述 将 ops-math 仓库中 19 个算子的 arch35 regbase kernel 实现从 Reg 层旧接口替换为新接口，同时将地址空间限定符 __local_mem__ 统一替换为 __ubuf__，使代码与 AscendC SDK 最新命名规范保持一致。 ### 改动原因 AscendC SDK 新版本对 Reg 层 API 进行了重命名，旧 API（如 Reg::DataCopy、Reg::CompareScalar、Reg::MaskAnd、Reg::MaskOr、Reg::MaskPack、Reg::FusedAbsSub、Reg::FusedMulDstAdd 等）将逐步废弃。本 PR 同步更新仓库中 arch35 regbase 算子 kernel 的接口调用，避免使用将废弃的 API。 ### 改动方法 对 22 个 arch35 regbase kernel 头文件进行批量接口替换，涉及 19 个 math 算子： add_n（2 个文件：add_n_regbase.h、add_n_regbase_fullload.h）、assign_add、assign_sub、cast（cast_impl.h）、ceil、exp、floor（3 个文件：floor_bf16.h、floor_f16.h、floor_f32.h）、is_close（is_close_dag.h）、log、log1p、nan_to_num、neg、reciprocal、round、rsqrt、rsqrt_grad、sign（sign.h）、tanh、trunc 接口替换映射如下： | 旧接口 | 新接口 | 说明 | |--------|--------|------| | Reg::DataCopy(reg, addr) | Reg::LoadAlign(reg, addr) | UB→Reg 加载 | | Reg::DataCopy(addr, reg) | Reg::StoreAlign(addr, reg) | Reg→UB 存储 | | Reg::DataCopy<T, LoadDist/StoreDist> | Reg::LoadAlign/StoreAlign<T, Dist> | 带 Dist 模板参数 | | Reg::DataCopy<T, PostLiteral, Dist> | Reg::LoadAlign/StoreAlign<T, PostLiteral, Dist> | 带 PostLiteral | | Reg::CompareScalar | Reg::Compares | 标量比较 | | CompareScalar（无命名空间） | Compares | 标量比较 | | Reg::MaskAnd | Reg::And | 掩码与运算 | | Reg::MaskOr | Reg::Or | 掩码或运算 | | Reg::MaskPack | Reg::Pack | 掩码打包 | | Reg::FusedAbsSub | Reg::AbsSub | 融合绝对差 | | Reg::FusedMulDstAdd | Reg::MulDstAdd | 融合乘加 | | __local_mem__ | __ubuf__ | 地址空间限定符（涉及 floor、is_close 算子） | > 注：上述替换同时覆盖带 AscendC:: 前缀和不带前缀两种调用形式。替换为纯命名变更，不涉及计算逻辑修改，部分接口替换伴随行格式重排（缩进调整）。 ## 关联的Issue https://gitcode.com/cann/ops-math/issues/2585 ## 测试 接口替换为纯命名变更，不涉及计算逻辑修改，通过编译验证和已有 ST/UT 用例回归验证。 ## 文档更新 无 ## 类型标签 - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [x] 其他，请描述：接口重命名，代码规范对齐 ## AI/Agent生成声明 - [x] AI辅助编写 See merge request: cann/ops-math!4527 | 2 months ago | |
math仓算子错误日志整改 Co-authored-by: qq_52874943<gaomingyuan3@h-partners.com> # message auto-generated for no-merge-commit merge: !5345 merge master into master math仓算子错误日志整改 Created-by: qq_52874943 Commit-by: qq_52874943 Merged-by: cann-robot Description: ## 描述 本 PR 针对 math 仓算子日志进行整改（关联 Issue #2853），主要目的是规范算子错误日志的级别与内容。改动覆盖约 35 个文件，核心内容包括三类：将 examples/test_geir_*.cpp 示例程序中 GE 初始化失败、加图失败、运行图失败、反初始化失败等错误场景的日志级别由 INFO 修正为 ERROR；修正 op_kernel_aicpu、op_api、tiling、framework 插件中参数校验及错误提示日志的英文语法与拼写；同时清理行尾多余空白字符。整体为纯日志与文档性整改，不涉及算子计算逻辑与对外接口行为变更。 ## 关联的Issue 关联Issue[#2853](https://gitcode.com/cann/ops-math/issues/2853) ## 测试 本地已完成新精度对比与性能测试，通过静态检查、蓝区冒烟、算子ST以及算子二级冒烟。 ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [x] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!5345 | 2 months ago | |
feat: Infershape整改，25个math算子InferDataType注册迁移至op_graph目录（IMPL_OP独立链）及UT看护 Co-authored-by: ywx1080114<yuanbin22@huawei.com> # message auto-generated for no-merge-commit merge: !6392 merge infershape-dtype into master feat: Infershape整改，25个math算子InferDataType注册迁移至op_graph目录（IMPL_OP独立链）及UT看护 Created-by: yuanbin_22 Commit-by: ywx1080114 Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动，包括改动的原因和所采取的方法。--> 作为 Infershape 整改的一部分，为 25 个 math 算子在 op_graph/ 目录新增 InferDataType（输出数据类型推导）独立注册及 UT 看护，共 50 个文件（+1757 -0，纯新增），对齐 ops-nn 仓方案（参考 PR#9839 样板）： **注册方案（IMPL_OP 独立链）：** - 每个算子新增 op_graph/<op>_graph_infer.cpp，通过 IMPL_OP(X).InferDataType(InferDataTypeX) 独立注册链注册，与 op_host 的 IMPL_OP_INFERSHAPE（InferShape）链解耦； - 涉及 25 个算子：abs_grad、add_n、assign_add、assign_sub、cast、cos、erf、exp、expm1、floor、log1p、logical_not、masked_scale、nan_to_num、neg、reciprocal、rsqrt、rsqrt_grad、sign、sin、sqrt、sqrt_grad、square、tanh、trunc； - 原 35 个算子中的另外 10 个（atan、ceil、is_finite、is_inf、is_nan、is_neg_inf、is_pos_inf、log、right_shift、round）的 op_graph 迁移已单独合入 master，不在本 PR 范围； **实现分类：** - 24 个算子输出 dtype 与第一个输入一致（SetOutputDataType(0, GetInputDataType(0))）； - Cast 按属性推导并**新增枚举范围校验**：从 gert::RuntimeAttrs 读取 dst_type 属性，校验合法值域为 [0, DT_MAX) 且排除 DT_UNDEFINED（=28），非法值在推导处直接报错返回 GRAPH_FAILED（与 arch31 tiling 的 TransAclToGeDataType 白名单校验同款防御，防止非法值静默写入输出 desc）。 **UT 看护（25 个测试文件，26 个用例）：** - 每算子新增 tests/ut/op_graph/test_<op>_graph_infer_pass.cpp，直接调用导出的 ops::InferDataTypeX 函数，通过 OpInferDataTypeContextBuilder 构造上下文，参数化断言推导返回 GRAPH_SUCCESS 且输出 dtype 符合预期； - Cast 参数化覆盖 dst_type 集合（含 FP16/FLOAT/BF16/INT 系列等），并新增非法值专项用例：dst_type 为 -1/DT_UNDEFINED/DT_MAX/999 时断言推导返回 GRAPH_FAILED。 ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。--> <!-- 如果这个PR是为了解决特定的问题单，请在这里描述问题单单号。--> - 关联 Issue #3792 ## 测试 <!--描述进行了哪些测试来验证你的改动。包括但不限于二级冒烟、算子泛化等。--> - 25 个算子各新增 InferDataTypeTest 参数化 UT 用例（直接函数调用），全部通过； - Cast 额外新增非法 dst_type 专项用例（-1/DT_UNDEFINED/DT_MAX/999 断言 GRAPH_FAILED），覆盖枚举校验防御路径。 ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> - 无需文档更新。 ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!6392 | 7 days ago | |
IsClose等15个算子新增支持 ascend960dt Co-authored-by: TH<taohai2@huawei.com> # message auto-generated for no-merge-commit merge: !6079 merge backBlue into master IsClose等15个算子新增支持 ascend960dt Created-by: TH_HW Commit-by: TH Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动，包括改动的原因和所采取的方法。--> 本次改动为 math 仓 15 个算子新增 Ascend960DT（ NpuArch::DAV_9201）平台支持，使这些算子可在 Ascend960DT 上完成注册、编译与运行。 适配算子：is_close、is_neg_inf、masked_scale、neg、reciprocal、round、rsqrt、rsqrt_grad、sign、sin、sqrt、sqrt_grad、square、tanh、trunc。 具体改动： - 算子注册：在各算子 op_host/*_def.cpp 中新增 this->AICore().AddConfig("ascend960dt", aicoreConfig)，完成 Ascend960DT 计算单元注册。 - 编译单元：在各算子 CMakeLists.txt 的 SUPPORT_COMPUTE_UNIT 中加入 ascend960dt，并按原有平台（arch22/arch35）在 SUPPORT_TILING_DIR 中补入对应的 arch35 tiling 目录。 - 二进制配置：新增 op_host/config/ascend960dt/<op>_binary.json，覆盖各算子的 dtype/shape 组合（如 float16/bfloat16/float32/int32/int8/uint8 等）；新增 <op>_simplified_key.ini 指定 --simplified_key_mode。 - 内核编译配置：在 scripts/kernel/binary_config/ascendc_config.json 中为上述算子的 compute_units 与 compile_options 增加 ascend960dt。 - 平台映射：在 scripts/kernel/binary_script/build_env.sh 的 SOC_MAP 中新增 [Ascend960DT]="Ascend960DT_968B7"。 - 算子校验：math/sign/op_api/aclnn_sign.cpp 的 dtype 校验新增 NpuArch::DAV_9201 分支，使 bf16 支持在 Ascend960DT 生效。 - 用例：补充 sin、sqrt 的 arch35 tiling UT（新增小 shape、大 shape、3D、非支持 dtype 等正向/异常用例）；sqrt 用例同时做格式化整改。 - 顺带统一部分 CMakeLists.txt 的许可头注释格式（add_n、complex_abs、cast、is_inf、log1p、tanh 等），并修正 neg 的 ascend950 simplified_key.ini 末尾换行。 ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。--> <!-- 如果这个PR是为了解决特定的问题单，请在这里描述问题单单号。--> NA ## 测试 <!--描述进行了哪些测试来验证你的改动。包括但不限于二级冒烟、算子泛化等。--> - 编译验证：执行 math 仓编包，确认 15 个算子在 Ascend960DT 下均能正常注册与编译，生成对应 kernel 二进制。 - 二进制/交付件验证：确认各算子 config/ascend960dt 下的 *_binary.json 与 *_simplified_key.ini 生效，ascendc_config.json 中已包含 ascend960dt 编译单元。 - 单元测试：执行 sin、sqrt 的 arch35 tiling UT，覆盖 fp16/bf16/fp32、小/大 shape、3D 及非支持 dtype 场景，结果符合预期。 ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> 无 ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!6079 | 21 days ago | |
docs: 修正 math 算子 README 文档中的描述与数据类型说明 Co-authored-by: Nice_try<zhangzhiwei108@huawei.com> # message auto-generated for no-merge-commit merge: !6158 merge 3246 into master docs: 修正 math 算子 README 文档中的描述与数据类型说明 Created-by: Nice_try Commit-by: Nice_try Merged-by: cann-robot Description: ## 描述 修正 math 目录下 13 个算子 README.md 文档中的描述问题，提升文档准确性与术语一致性，具体包括： - 数据类型说明修正： abs 输出类型列表去掉重复的 FLOAT16、FLOAT，改为 COMPLEX32、COMPLEX64；assign_sub 修正拼写 BFOAT16 → BFLOAT16。 - 属性名拼写修正：hans_encode 的 reshuff → reshuffle。 - 术语大小写统一：arg_min 的 infershape → InferShape；reduce_all 的 tenor → Tensor。 - 表述/用词修正：add_n 的"主元素相加"→"逐元素相加"；add_lora 的"Lora"→"LoRa"；cholesky 的"需为"→"须为"；cosine_similarity 的"显示指定"→"显式指定"；cumsum 的"累计求和"→"累积求和"；lgamma 的"输入张量x1"→"输入张量x"；mul_addn 表述精简；asinh_grad 的"通过图模式方式调用"→"通过图模式调用"。 ### 改动原因 文档中存在错别字、术语拼写与大小写不统一、数据类型说明错误等问题，影响可读性与准确性。 ### 改动方法 逐文件校对并修正上述 README.md 中的文本与类型说明，仅涉及文档，不改变任何代码逻辑。 ## 关联的Issue #3426 ## 测试 文档变更，无需代码测试。 ## 文档更新 更新 math 目录下以下 13 个 README.md： math/abs、math/add_lora、math/add_n、math/arg_min、math/asinh_grad、math/assign_sub、math/cholesky、math/cosine_similarity、math/cumsum、math/hans_encode、math/lgamma、math/mul_addn、math/reduce_all ## 类型标签 - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [x] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!6158 | 1 month ago |
AddN
产品支持情况
| 产品 | 是否支持 |
|---|---|
| Ascend 950PR&950DT系列产品 | √ |
| Atlas A3系列产品 | √ |
| Atlas A2系列产品 | √ |
| Atlas 200I/500 A2推理产品 | × |
| Atlas推理系列产品 | × |
| Atlas训练系列产品 | √ |
功能说明
-
算子功能：对输入进行逐元素相加求和操作。
-
计算公式：
y=x1+x2+...+xny = x_1 + x_2+ ... +x_n y=x1+x2+...+xn
参数说明
| 参数名 | 输入/输出/属性 | 描述 | 数据类型 | 数据格式 |
|---|---|---|---|---|
| x | 输入 | 公式中的输入张量x | INT32, INT64, FLOAT16, BFLOAT16, FLOAT32 | ND |
| y | 输出 | 公式中的输出张量y | 同x | ND |
| N | 属性 | 公式中的输入x的数量 | INT32 | ND |
调用说明
| 调用方式 | 调用样例 | 说明 |
|---|---|---|
| 图模式调用 | test_geir_add_n | 通过算子IR构图方式调用AddN算子 |
| aclnn调用 | test_aclnn_add_n | 通过aclnn API方式调用AddN算子 |