cann-robotfeat: Infershape整改，25个math算子InferDataType注册迁移至op_graph目录（IMPL_OP独立链）及UT看护

Files	Last commit	Last update
examples	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
framework	迁移开源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
op_graph	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
op_host	math仓121算子迁蓝 Co-authored-by: andong_hw<andong2@huawei.com> # message auto-generated for no-merge-commit merge: !5762 merge master into master math仓121算子迁蓝 Created-by: qq_52874943 Commit-by: andong_hw Merged-by: cann-robot Description: ## 描述 math仓部分121算子黄迁蓝，只进行960dt适配，以及算子二进制补充，不动kernel和tiling代码。 ## 关联的Issue 关联Issue[#3650](https://gitcode.com/cann/ops-math/issues/3650) ## 测试本地已完成新精度对比与性能测试，通过静态检查、蓝区冒烟、算子ST以及算子二级冒烟。 ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!5762	22 days ago
op_kernel	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
tests	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
CMakeLists.txt	math仓121算子迁蓝 Co-authored-by: andong_hw<andong2@huawei.com> # message auto-generated for no-merge-commit merge: !5762 merge master into master math仓121算子迁蓝 Created-by: qq_52874943 Commit-by: andong_hw Merged-by: cann-robot Description: ## 描述 math仓部分121算子黄迁蓝，只进行960dt适配，以及算子二进制补充，不动kernel和tiling代码。 ## 关联的Issue 关联Issue[#3650](https://gitcode.com/cann/ops-math/issues/3650) ## 测试本地已完成新精度对比与性能测试，通过静态检查、蓝区冒烟、算子ST以及算子二级冒烟。 ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!5762	22 days ago
README.md	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

AssignSub

产品支持情况

产品	是否支持
Ascend 950PR&950DT系列产品	√
Atlas A3系列产品	√
Atlas A2系列产品	√
Atlas 200I/500 A2推理产品	×
Atlas推理系列产品	√
Atlas训练系列产品	√

功能说明

算子功能：完成在原有tensor上的减法计算。
计算公式：

out_i = var_i - value_i

参数说明

参数名	输入/输出/属性	描述	数据类型	数据格式
var	输入	输入张量，公式中的var_i。	BFLOAT16、FLOAT16、FLOAT、INT8、INT32、INT64、UINT8	ND
value	输入	输入张量，表示要从var_i上减去的值，公式中的out_i。	BFLOAT16、FLOAT16、FLOAT、INT8、INT32、INT64、UINT8	ND
var	输出	输出张量，公式中的out_i,与var_i同地址。	BFLOAT16、FLOAT16、FLOAT、INT8、INT32、INT64、UINT8	ND
use_locking	属性	可选，是否使用锁来保护更新操作。	bool	ND

约束说明

无

调用说明

调用方式	调用样例	说明
图模式调用	test_geir_assign_sub.cpp	通过算子IR构图方式调用AssignSub算子。