Star0

Star0

cann-robot常量折叠和infershape整改

c257cedf创建于 10 天前历史提交

文件	最后提交记录	最后更新时间
examples	fix(geir-test): 修复 op-math geir 测试日志级别不一致（INFO→ERROR） Co-authored-by: sun-yibo1<sunyibo1@hisilicon.com> # message auto-generated for no-merge-commit merge: !4936 merge fix-geir-test-log-level-mismatch into master fix(geir-test): 修复 op-math geir 测试日志级别不一致（INFO→ERROR） Created-by: sun-yibo1 Commit-by: sun-yibo1 Merged-by: cann-robot Description: ## 描述修复 op-math 仓库 17 个 geir 测试文件中日志级别与描述不一致的问题：失败消息被误标记为 `INFO`，而同文件内其他失败消息均正确标记为 `ERROR`，构成内部不一致。本次将失败消息的日志级别从 `INFO` 改为 `ERROR`。 ### 改动原因 geir 测试统一使用 `printf("%s - LEVEL - [XIR]: message")` 格式。以下失败分支的消息明确表示失败，但日志级别却写成 `INFO`，与同文件内其他失败消息（如 `Generate input data failed`、`Create ir session using build options failed`，均标记为 `ERROR`）不一致，会误导日志排查： - `Initialize ge using ge global options failed` - `Run graph failed` - `Finalize ir graph session failed` - `Add graph failed`（部分文件） ### 改动方法 - 将上述失败消息的日志级别标签 `- INFO -` 改为 `- ERROR -`，不修改任何测试逻辑、消息文本或参数。 - 同时按 pre-commit `clang-format` 要求对文件做整体格式化：这些文件在 upstream 上即不满足仓库 `.clang-format` 规范，pre-commit 的 `clang-format` hook 会强制格式化整个文件才能通过，故本次一并完成格式化（纯格式，无语义变更）。 ### 涉及算子（17 个 geir 测试文件，均为非 experimental 目录） add、sub、mul、div、real_div、pow、less、greater、equal、not_equal、clip_by_value、clip_by_value_v2、bias_add、squared_difference、tanh_grad、axpy、axpy_v2、select_v2。 ### 说明 - 仅日志级别与格式修改，不影响测试编译与运行结果。 - pre-commit 全部 hook 通过（trailing-whitespace / end-of-file-fixer / clang-format / codespell / check-* / detect-private-key）。OAT 合规检查因本机 oat-py 2.0.0 + Python 3.8 环境对 OAT.xml 的 copyright 策略解析存在已知问题（对所有文件均误报 "Copyright Header Invalid"，包括完全标准 header 的非 geir 文件），无法在本地通过，已用 `--no-verify` 提交；该问题是仓库级工具/配置问题，非本 PR 引入。 ## 关联的Issue - #2792 ## 测试 - 仅日志级别/格式修改，无逻辑变更，不影响测试编译与运行结果。 - 通过 `git diff` 逐项核对：失败消息的 `- INFO -` → `- ERROR -`，消息文本与参数不变。 - pre-commit 全部 hook 通过（除 OAT 仓库级工具问题，详见描述）。 ## 文档更新无。 ## 类型标签 - [x] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!4936	11 天前
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	1 个月前
op_graph	math仓算子代码整改 Co-authored-by: cai-chengchao<caichengchao1@h-partners.com> # message auto-generated for no-merge-commit merge: !873 merge math_op_back_2 into master math仓算子代码整改 Created-by: cai-chengchao Commit-by: cai-chengchao Merged-by: cann-robot Description: ## 描述 math仓算子代码整改 ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。--> <!-- 如果这个PR是为了解决特定的问题单，请在这里描述问题单单号。--> ## 测试 <!--描述进行了哪些测试来验证你的改动。包括但不限于二级冒烟、算子泛化等。--> ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!873	7 个月前
op_host	refactor: 消除 math 算子重复头文件依赖（下），统一引用 base 仓 Co-authored-by: tianqiguang<tianqiguang@huawei.com> # message auto-generated for no-merge-commit merge: !2980 merge math_tiling into master refactor: 消除 math 算子重复头文件依赖（下），统一引用 base 仓 Created-by: tianqiguang Commit-by: tianqiguang Merged-by: cann-robot Description: ## 描述消除 math 目录下后半段（ger ~ zero_op）共 89 个算子的重复头文件依赖，将 tiling 基类和模板注册引用统一到 base 仓。本 PR 与 #2969 共同覆盖全部 math 算子。 ### 改动原因与 #2969 一致，剩余 math 算子（ger 起）各自直接引用旧头文件并存在全局命名空间引入： - tiling_base.h 中 TilingBaseClass 已在 base 仓提取为 tiling_base_class.h - tiling_templates_registry.h 中 math 注册宏已拆分到 math_tiling_templates_registry.h - using namespace Ops::Math::OpTiling 引入全局命名空间，存在符号冲突风险 ### 改动方法批量更新 math 算子文件的 #include 及命名空间引用： - tiling_base.h -> tiling_base_class.h - tiling_templates_registry.h -> math_tiling_templates_registry.h - 移除 using namespace Ops::Math::OpTiling - TilingBaseClass -> Ops::Base::TilingBaseClass - TilingRegistry::GetInstance() -> Ops::Math::OpTiling::TilingRegistry::GetInstance() ## 关联的Issue - #1685 ## 测试 - 涉及的 math 算子二级冒烟测试已通过 ## 文档更新无文档变更 ## 类型标签 - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [x] 其他，请描述：代码重构，消除重复头文件依赖 See merge request: cann/ops-math!2980	3 个月前
op_kernel	abs等算子A5实现 Co-authored-by: ly_cann_coder<liyao152@huawei.com> # message auto-generated for no-merge-commit merge: !495 merge develop into master abs等算子A5实现 Created-by: ly_cann_coder Commit-by: ly_cann_coder Merged-by: cann-robot Description: ## 描述新增以下算子的A5实现 \|序号\| 算子 \| 功能 \| \|--\|--\|--\| \|1\| abs \| 为输入张量的每一个元素取绝对值 \| \|2\| add \| 对输入完成相加操作 \| \|3\| addcdiv \| 张量运算函数，用于执行乘除加组合操作，将张量除法（带缩放）+ 张量加法合并为单个操作 \| \|4\| addcmul \| 张量运算函数，用于执行乘除加组合操作，将张量乘法（带缩放）+ 张量加法合并为单个操作 \| \|5\| axpy \| 源操作数中每个元素与标量求积后和目的操作数中的对应元素相加 \| \|6\| bincount \| 计算非负整数数组中每个数的频率 \| \|7\| ceil \| 返回输入tensor中每个元素向上取整的结果 \| \|8\| clip_by_value_v2 \| 将输入的所有元素限制在一定范围内 \| \|9\| concat \| 用于沿指定维度将多个输入 Tensor 进行拼接，输出包含所有输入数据按顺序拼接后的 Tensor \| \|10\| concat_v2 \| 用于沿指定维度将多个输入 Tensor 进行拼接，输出包含所有输入数据按顺序拼接后的 Tensor \| \|11\| equal \| 计算两个Tensor是否有相同的大小和元素，返回一个Bool类型 \| \|12\|exp\|返回一个新的张量，该张量的每个元素都是输入张量对应元素的指数\| \|13\| fill \| 对张量进行填充操作，支持非连续的Tensor操作 \| \|14\| floor \| 返回输入Tensor中每个元素向下取整，并将结果回填到输入Tensor中 \| \|15\| floor_div \| 完成除法计算，对余数向下取整 \| \|16\| floor_mod \| 用于计算两个数的模 \| \|17\| is_neg_inf \| 判断输入张量的元素是否为负无穷 \| \|18\| is_pos_inf \| 判断输入张量的元素是否为正无穷 \| \|19\| less \| 判断输入self中的每个元素是否小于输入other的值，返回一个Bool类型的Tensor \| \|20\| less_equal \| 判断输入self中的元素值是否小于等于other的值，并将self的每个元素的值与other值的比较结果写入out中 \| \|21\| logical_and \| 对两个输入张量的对应元素执行「与逻辑」判断，输出布尔型张量（True/False） \| \|22\| log \| 对输入张量x的元素，逐元素进行对数计算，并将结果保存到输出张量y中 \| \|23\| not_equal \| 逐元素比较两个输入张量是否不相等 \| \|24\| ones_like \| 返回形状和类型相同的张量，所有元素都设置为1 \| \|25\|sign\|对输入的tensor逐元素进行Sign符号函数的运算并输出结果tensor\| \|26\| sqrt \| 完成非负数平方根计算，负数情况返回nan \| \|27\| sqrt_grad \| 完成梯度平方根计算 \| \|28\| square \| 为输入张量的每一个元素计算平方值 \| \|29\| squared_difference \| 为第一个输入张量减去第二个输入张量，并计算其平方值 \| \|30\| tanh \| 激活函数，返回与输入tensor shape相同的tensor，对输入tensor进行elementwise的计算 \| \|31\| tril \| 将输入的self张量的最后二维（按shape从左向右数）沿对角线的右上部分置零 \| \|32\| triu \| 将输入的self张量的最后二维（按shape从左向右数）沿对角线的左下部分置零 \| \|33\| zeros_like \| 将张量填充为全零 \| ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。--> <!-- 如果这个PR是为了解决特定的问题单，请在这里描述问题单单号。--> ## 测试 <!--描述进行了哪些测试来验证你的改动。包括但不限于二级冒烟、算子泛化等。--> ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [x] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!495	8 个月前
op_kernel_aicpu	常量折叠和infershape整改 Co-authored-by: liu-lu<www.liulu824910939@qq.com> # message auto-generated for no-merge-commit merge: !4582 merge master into master 常量折叠和infershape整改 Created-by: liu-lu Commit-by: liu-lu Merged-by: cann-robot Description: ## 描述本 PR 为 6 个 AICPU 算子（add、round、search_sorted、select_v2、square、squared_difference）使能 GE 图编译期常量折叠，并对 search_sorted、select_v2 进行 infershape/OpDef 整改。 ### 变更内容 1. 常量折叠使能（6 算子） \| 算子 \| CMakeLists \| kernel 注册 \| \|------\|-----------\|-------------\| \| add \| 增加 `HOSTCPU TRUE` \| `REGISTER_CPU_KERNEL` → `OPS_MATH_REGISTER_CPU_KERNELV2` \| \| round \| 增加 `HOSTCPU TRUE` \| `REGISTER_CPU_KERNEL` → `OPS_MATH_REGISTER_CPU_KERNELV2` \| \| search_sorted \| 增加 `HOSTCPU TRUE` \| `REGISTER_CPU_KERNEL` → `OPS_MATH_REGISTER_CPU_KERNELV2` \| \| select_v2 \| 增加 `HOSTCPU TRUE` \| `REGISTER_CPU_KERNEL` → `OPS_MATH_REGISTER_CPU_KERNELV2` \| \| square \| 增加 `HOSTCPU TRUE` \| `REGISTER_CPU_KERNEL` → `OPS_MATH_REGISTER_CPU_KERNELV2` \| \| squared_difference \| 增加 `HOSTCPU TRUE` \| `REGISTER_CPU_KERNEL` → `OPS_MATH_REGISTER_CPU_KERNELV2` \| - 各 kernel 增加 `#include "aicpu/math_aicpu_register.h"`； - CMakeLists 增加 `HOSTCPU TRUE`，将算子编入 host 侧折叠 so（libopconstant_folding_math.so）。 2. infershape / OpDef 整改 - search_sorted：新增 `op_host/search_sorted_infershape.cpp`，实现 `InferShape4SearchSorted`，输出 shape 取 values（input1）的 shape； - select_v2：新增 `op_kernel_aicpu/select_v2_aicpu_def.cpp`（基于 OpDef 注册输入输出数据类型），删除旧 `select_v2.json`（JSON 配置方式），统一为 OpDef 注册链路。 ## 关联的Issue #2847 ## 测试根据代码变更，测试场景如下： 1. 编译构建测试 - 执行 cmake 配置与编译，验证 6 个算子 CMakeLists `HOSTCPU TRUE` 配置变更正确； - 确认 `libopconstant_folding_math.so` 中已编入上述 6 个算子的 host 侧 kernel。 2. 常量折叠验证 - 输入为常量时，验证算子可在图编译期被折叠求值（host 侧执行），运行期不再下发 device kernel； - 输入为非常量时，验证 device 侧执行逻辑与折叠前保持一致。 3. InferShape 验证（search_sorted） - 验证 search_sorted 输出 shape 与 values 输入 shape 一致； - 覆盖多维度、边界 shape 场景。 4. 算子注册/加载验证（select_v2） - 验证 select_v2 基于 OpDef 注册后接口可正确加载与调用； - 覆盖全数据类型（COMPLEX128/COMPLEX64/DOUBLE/FLOAT/FLOAT16/INT16/INT32/INT64/INT8/UINT16/UINT32/UINT64/UINT8/BOOL）。 5. 算子泛化 - 对 6 个算子执行二级冒烟与泛化用例，确认功能正确性不受影响。 ## 文档更新本 PR 不涉及 README/aclnn 接口文档更新。 ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [x] 新特性 - [x] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!4582	10 天前
tests	新增部分用例的golden Co-authored-by: 鲜磊<xianlei4@h-partners.com> # message auto-generated for no-merge-commit merge: !3262 merge g3 into master 新增部分用例的golden Created-by: xianlei2 Commit-by: 鲜磊 Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动，包括改动的原因和所采取的方法。--> ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。--> <!-- 如果这个PR是为了解决特定的问题单，请在这里描述问题单单号。--> ## 测试 <!--描述进行了哪些测试来验证你的改动。包括但不限于二级冒烟、算子泛化等。--> ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!3262	2 个月前
CMakeLists.txt	常量折叠和infershape整改 Co-authored-by: liu-lu<www.liulu824910939@qq.com> # message auto-generated for no-merge-commit merge: !4582 merge master into master 常量折叠和infershape整改 Created-by: liu-lu Commit-by: liu-lu Merged-by: cann-robot Description: ## 描述本 PR 为 6 个 AICPU 算子（add、round、search_sorted、select_v2、square、squared_difference）使能 GE 图编译期常量折叠，并对 search_sorted、select_v2 进行 infershape/OpDef 整改。 ### 变更内容 1. 常量折叠使能（6 算子） \| 算子 \| CMakeLists \| kernel 注册 \| \|------\|-----------\|-------------\| \| add \| 增加 `HOSTCPU TRUE` \| `REGISTER_CPU_KERNEL` → `OPS_MATH_REGISTER_CPU_KERNELV2` \| \| round \| 增加 `HOSTCPU TRUE` \| `REGISTER_CPU_KERNEL` → `OPS_MATH_REGISTER_CPU_KERNELV2` \| \| search_sorted \| 增加 `HOSTCPU TRUE` \| `REGISTER_CPU_KERNEL` → `OPS_MATH_REGISTER_CPU_KERNELV2` \| \| select_v2 \| 增加 `HOSTCPU TRUE` \| `REGISTER_CPU_KERNEL` → `OPS_MATH_REGISTER_CPU_KERNELV2` \| \| square \| 增加 `HOSTCPU TRUE` \| `REGISTER_CPU_KERNEL` → `OPS_MATH_REGISTER_CPU_KERNELV2` \| \| squared_difference \| 增加 `HOSTCPU TRUE` \| `REGISTER_CPU_KERNEL` → `OPS_MATH_REGISTER_CPU_KERNELV2` \| - 各 kernel 增加 `#include "aicpu/math_aicpu_register.h"`； - CMakeLists 增加 `HOSTCPU TRUE`，将算子编入 host 侧折叠 so（libopconstant_folding_math.so）。 2. infershape / OpDef 整改 - search_sorted：新增 `op_host/search_sorted_infershape.cpp`，实现 `InferShape4SearchSorted`，输出 shape 取 values（input1）的 shape； - select_v2：新增 `op_kernel_aicpu/select_v2_aicpu_def.cpp`（基于 OpDef 注册输入输出数据类型），删除旧 `select_v2.json`（JSON 配置方式），统一为 OpDef 注册链路。 ## 关联的Issue #2847 ## 测试根据代码变更，测试场景如下： 1. 编译构建测试 - 执行 cmake 配置与编译，验证 6 个算子 CMakeLists `HOSTCPU TRUE` 配置变更正确； - 确认 `libopconstant_folding_math.so` 中已编入上述 6 个算子的 host 侧 kernel。 2. 常量折叠验证 - 输入为常量时，验证算子可在图编译期被折叠求值（host 侧执行），运行期不再下发 device kernel； - 输入为非常量时，验证 device 侧执行逻辑与折叠前保持一致。 3. InferShape 验证（search_sorted） - 验证 search_sorted 输出 shape 与 values 输入 shape 一致； - 覆盖多维度、边界 shape 场景。 4. 算子注册/加载验证（select_v2） - 验证 select_v2 基于 OpDef 注册后接口可正确加载与调用； - 覆盖全数据类型（COMPLEX128/COMPLEX64/DOUBLE/FLOAT/FLOAT16/INT16/INT32/INT64/INT8/UINT16/UINT32/UINT64/UINT8/BOOL）。 5. 算子泛化 - 对 6 个算子执行二级冒烟与泛化用例，确认功能正确性不受影响。 ## 文档更新本 PR 不涉及 README/aclnn 接口文档更新。 ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [x] 新特性 - [x] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!4582	10 天前
README.md	doc Tools扫描错误修改 Co-authored-by: gitee-yanglulu<yanglulul@h-partners.com> # message auto-generated for no-merge-commit merge: !1899 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!1899	5 个月前

SquaredDifference

产品支持情况

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

功能说明

算子功能：为第一个输入张量减去第二个输入张量，并计算其平方值。
计算公式：
$y = (x1 - x2) * (x1 - x2) = (x1 - x2)^2$

参数说明

参数名	输入/输出/属性	描述	数据类型	数据格式
x1	输入1	公式中的输入张量x1。	BFLOAT16、FLOAT16、FLOAT、INT32、INT64	ND
x2	输入2	公式中的输入张量x2。	BFLOAT16、FLOAT16、FLOAT、INT32、INT64	ND
y	输出	公式中的输出张量y。	BFLOAT16、FLOAT16、FLOAT、INT32、INT64	ND

Atlas 训练系列产品、Atlas 推理系列产品：不支持BFLOAT16。

调用说明

调用方式	调用样例	说明
图模式调用	test_geir_squared_difference	通过算子IR构图方式调用Square算子。