Star0

Star0

cann-robotfeat: 算子910B~910E区间判断追加IsRegBase()并统一DAV_3510硬编码判断以兼容后续Regbase芯片

82321099创建于 5 天前历史提交

文件	最后提交记录	最后更新时间
docs	资料问题整改 Co-authored-by: weike<weike13@huawei.com> # message auto-generated for no-merge-commit merge: !5429 merge master into master 资料问题整改 Created-by: m0_55003149 Commit-by: weike Merged-by: cann-robot Description: ## 描述 1、math/sub/docs/aclnnSub&aclnnInplaceSub.md，代码示例中打印信息拼写错误 2、math/erfc/docs/aclnnErfc&aclnnInplaceErfc.md，宏参数名'return_efr'拼写错误 ## 关联的Issue https://gitcode.com/cann/ops-math/issues/3077 ## 测试 <!--描述进行了哪些测试来验证你的改动。包括但不限于二级冒烟、算子泛化等。--> ## 文档更新 1、math/sub/docs/aclnnSub&aclnnInplaceSub.md 2、math/erfc/docs/aclnnErfc&aclnnInplaceErfc.md ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [x] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!5429	8 天前
examples	修改argmin/cumprod等部分算子的aclnn文档 example实例 Co-authored-by: jiangjiawei<jiangjiawei10@huawei.com> # message auto-generated for no-merge-commit merge: !5423 merge aclnn_example_0907 into master 修改argmin/cumprod等部分算子的aclnn文档 example实例 Created-by: jiangjiawei Commit-by: jiangjiawei Merged-by: cann-robot Description: ## 描述修改argmin/cumprod等部分算子的aclnn文档 example实例 ## 关联的Issue https://gitcode.com/cann/ops-math/issues/3084 ## 测试文档修改，不涉及 ## 文档更新见PR ## 类型标签 <!-- [x] 表示选中 --> - [x] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!5423	8 天前
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_api	feat: 算子910B~910E区间判断追加IsRegBase()并统一DAV_3510硬编码判断以兼容后续Regbase芯片 Co-authored-by: hahaha22<wangrunze20@h-partners.com> # message auto-generated for no-merge-commit merge: !5467 merge q_aclnn_version into master feat: 算子910B~910E区间判断追加IsRegBase()并统一DAV_3510硬编码判断以兼容后续Regbase芯片 Created-by: hahaha22 Commit-by: hahaha22 Merged-by: cann-robot Description: ## 描述以 RegBase 架构（NpuArch = DAV_3510，含 Ascend950 及后续 Regbase 芯片）兼容为目标，统一两类芯片判断写法： 1. 区间判断追加：对 13 个算子 op_api 中 `GetSocVersion() >= ASCEND910B && GetSocVersion() <= ASCEND910E` 区间判断追加 `\|\| IsRegBase()`，使 RegBase 平台在 SocVersion 区间之外时也复用 910B 支持分支（放开 BF16 dtype 校验） 2. 硬编码统一：将 11 个算子文件中 `case NpuArch::DAV_3510:` 硬编码判断改为 `IsRegBase()` 谓词，后续 Regbase 芯片（IsRegBase 集合扩展新 arch 值时）无需逐文件修改即可自动命中 ## 修改内容一：区间判断追加（13 个文件，15 处） \| 算子 \| 文件 \| 修改位置 \| \|------\|------\|----------\| \| less \| math/less/op_api/aclnn_lt_scalar.cpp \| GetDtypeSupportList、GetOutDtypeSupportList（2 处） \| \| less \| math/less/op_api/aclnn_lt_tensor.cpp \| GetDtypeSupportList、GetOutDtypeSupportList（2 处） \| \| less_equal \| math/less_equal/op_api/aclnn_le_scalar.cpp \| GetOutputDtypeSupportList（1 处） \| \| logical_or \| math/logical_or/op_api/aclnn_logical_or.cpp \| CheckSocVersionIsSupportBf16（1 处） \| \| maximum \| math/maximum/op_api/aclnn_maximum.cpp \| GetDtypeSupportList（1 处） \| \| minimum \| math/minimum/op_api/aclnn_minimum.cpp \| GetDtypeSupportList（1 处） \| \| mul \| math/mul/op_api/aclnn_mul.cpp \| GetDtypeSupportList（1 处） \| \| not_equal \| math/not_equal/op_api/aclnn_logical_xor.cpp \| CheckSocVersionIsSupportBf16（1 处） \| \| pow \| math/pow/op_api/aclnn_pow.cpp \| CheckSocVersionIsSupportBf16（1 处） \| \| pow \| math/pow/op_api/aclnn_pow_tensor_tensor.cpp \| CheckSocVersionIsSupportBf16（1 处） \| \| signbit \| math/signbit/op_api/aclnn_signbit.cpp \| GetDtypeSupportList（1 处） \| \| sub \| math/sub/op_api/sub.cpp \| IsAiCoreSupport（1 处） \| \| equal \| math/tensor_equal/op_api/aclnn_equal.cpp \| CheckSocVersionGe910B（1 处） \| 其中 sub.cpp、aclnn_pow_tensor_tensor.cpp、aclnn_logical_xor.cpp、aclnn_equal.cpp 补充 `#include "op_api/aclnn_check.h"`。 pow.cpp 的 IsAiCoreSupport、aclnn_ne_scalar.cpp、aclnn_ne_tensor.cpp：区间判断前已有 `if (IsRegBase())` 提前返回分支（追加会造成死条件），经评审不修改。 ## 修改内容二：硬编码判断统一（11 个文件，13 处） ### A类：switch 双 case（DAV_3510 与 DAV_2201 同分支，8处） switch 前置 `if (IsRegBase(curArch))` 返回原 DAV_3510 分支列表，删除 `case NpuArch::DAV_3510:` fall-through 行： - Add：math/add/op_api/add.cpp、aclnn_add.cpp - Sub：math/sub/op_api/aclnn_sub.cpp、aclnn_rsub.cpp - div：math/div/op_api/div.cpp（补 include） - floordiv：math/floor_div/op_api/floordiv.cpp（补 include） - tanhgrad：math/tanh_grad/op_api/aclnn_tanh_backward.cpp - floormod：math/floor_mod/op_api/aclnn_remainder.cpp ### B类：switch 独立 case（DAV_3510 单独返回 REGBASE 专用列表，3处） switch 前置 `if (IsRegBase(curArch))` 返回 REGBASE 列表，删除独立 case 块： - mul：math/mul/op_api/mul.cpp（REGBASE_AICORE_DTYPE_SUPPORT_LIST） - equal：math/equal/op_api/aclnn_eq_tensor.cpp、aclnn_eq_scalar.cpp（REGBASE_DTYPE_SUPPORT_LIST） ## 关联的Issue - #3100 （math仓910B~910E区间判断的RegBase兼容遗留清理：isclose恒真条件bug、8个同类判断文件未覆盖IsRegBase） ## 测试 - pre-commit 检查（clang-format/codespell/OAT 等）全部通过（含对修改文件的全量格式化） - 全部修改文件 g++ -fsyntax-only 语法验证通过 - 已触发 compile 编译 - 建议测试项： 1. RegBase（DAV_3510）平台验证上述算子 BF16 输入/输出 dtype 校验与 910B 平台一致 2. Ascend950 平台回归 A/B 类算子 dtype 选择行为不变 3. 回归 910B~910E、910/1980、310/310P 平台行为不受影响 ## 文档更新无 ## 类型标签 - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： ### 关联issue - [cann/ops-math#5512（报错 dtype 支持列表双层中括号修复，同改 aclnn_remainder.cpp/eq 系列）](https://gitcode.com/cann/ops-math/pull/5512) - [cann/ops-math#5519（RemainderScalarTensor Regbase 计算类型校验，同改 aclnn_remainder.cpp）](https://gitcode.com/cann/ops-math/pull/5519) - [cann/ops-nn#10101（ops-nn 侧同类 Regbase 判断统一）](https://gitcode.com/cann/ops-nn/pull/10101) See merge request: cann/ops-math!5467	5 天前
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	concat/pad/slice等算子支持新平台 Co-authored-by: ASCEND222<dongfei16@h-partners.com> # message auto-generated for no-merge-commit merge: !5184 merge master into master concat/pad/slice等算子支持新平台 Created-by: ASCEND222 Commit-by: ASCEND222 Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动，包括改动的原因和所采取的方法。--> 算子支持新平台 ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。--> <!-- 如果这个PR是为了解决特定的问题单，请在这里描述问题单单号。--> https://gitcode.com/cann/ops-math/issues/2992 ## 测试 <!--描述进行了哪些测试来验证你的改动。包括但不限于二级冒烟、算子泛化等。--> ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!5184	15 天前
op_kernel	masked_fill等算子下一代实现 Co-authored-by: ly_cann_coder<liyao152@huawei.com> # message auto-generated for no-merge-commit merge: !519 merge local into master masked_fill等算子下一代实现 Created-by: ly_cann_coder Commit-by: ly_cann_coder Merged-by: cann-robot Description: ## 描述新增以下算子的下一代实现： \| 序号 \| 名称 \| 功能 \| \|--\|--\|--\| \|1\| add_n \| 将多个输入tensor相加 \| \|2\|axpy_v2\|执行标量乘向量加向量的操作\| \|3\|lerp\|根据给定的权重，在起始和结束Tensor之间进行线性插值，返回插值后的Tensor\| \|4\|logical_or\|完成给定输入张量元素的逻辑或运算\| \|5\|masked_fill\|将输入Tensor`x`中mask位置为`True`的元素填充指定的值\| \|6\|sub\|完成减法计算，被减数按alpha进行缩放\| \|7\|addr\|求一维向量vec1和vec2的外积得到一个二维矩阵，并将外积结果矩阵乘一个系数后和自身乘系数相加后输出\| \|8\|square_sum_v1\|用于计算输入张量的平方和\| \|9\|is_nan\|判断张量中哪些元素是nan\| \|10\|assign_add\|完成在原有tensor上的加法操作\| \|11\|assign_sub\|完成在原有tensor上的减法操作\| \|12\|invert\|按位取反\| ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。--> <!-- 如果这个PR是为了解决特定的问题单，请在这里描述问题单单号。--> ## 测试 <!--描述进行了哪些测试来验证你的改动。包括但不限于二级冒烟、算子泛化等。--> ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [x] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!519	8 个月前
op_kernel_aicpu	feat: migrate math+conversion json ops to opdef pipeline Co-authored-by: zhaowenrui666<zhaowenrui7@huawei.com> # message auto-generated for no-merge-commit merge: !2130 merge mathdef into master feat: migrate math+conversion json ops to opdef pipeline Created-by: zhaowenrui666 Commit-by: zhaowenrui666 Merged-by: cann-robot Description: ## 描述 math仓aicpu算子信息库由json融合修改为op def文件，修改注册方式以及工程适配 ## 关联的Issue https://gitcode.com/cann/ops-math/issues/1271 ## 测试本次按 ENABLE_CUSTOM=ON 的 custom 路径验收，已验证 OpDef -> libaicpu_ops.so -> op_build --aicpu -> aicpu_kernel.ini -> parser_ini.py -> cust_aicpu_kernel.json 链路跑通，相关产物已生成并校验通过。AICPU before/after 与 I/O 对比已按新规则重评：input/output -> input0/output0 视为规范化差异；RightShift、SearchSorted、Sinh 在整改前 JSON 缺 I/O，但已对齐 CANN 基线 aicpu_kernel.ini，不判缺失；AddN: input0 -> dynamic_input0 和 AddFake -> Add 也已确认接受，不作为阻塞项。 AICore 回归已补做，before/after 经 parser_ini.py 转 json 后无差异。综合结论：本提交满足当前 OpDef 整改验收标准，可通过。 ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!2130	4 个月前
tests	add aclnn golden Co-authored-by: gcw_CRjbEjwl<fengdaoyong@h-partners.com> # message auto-generated for no-merge-commit merge: !4593 merge 20260812 into master add aclnn golden Created-by: gcw_CRjbEjwl Commit-by: gcw_CRjbEjwl Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动，包括改动的原因和所采取的方法。--> 算子增加aclnn的golden注册函数，由冒烟流水调用，跑新版ttk用例。 ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。例如：关联Issue #000--> 不涉及 ## 测试 <!--描述进行了哪些测试来验证你的改动。--> 本地完成验证 ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> 不涉及 ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [x] 其他，请描述： See merge request: cann/ops-math!4593	20 天前
CMakeLists.txt	concat/pad/slice等算子支持新平台 Co-authored-by: ASCEND222<dongfei16@h-partners.com> # message auto-generated for no-merge-commit merge: !5184 merge master into master concat/pad/slice等算子支持新平台 Created-by: ASCEND222 Commit-by: ASCEND222 Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动，包括改动的原因和所采取的方法。--> 算子支持新平台 ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。--> <!-- 如果这个PR是为了解决特定的问题单，请在这里描述问题单单号。--> https://gitcode.com/cann/ops-math/issues/2992 ## 测试 <!--描述进行了哪些测试来验证你的改动。包括但不限于二级冒烟、算子泛化等。--> ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!5184	15 天前
README.md	md大模型检测低错修改 Co-authored-by: gitcode-chenjiao<chenjiao31@huawei.com> # message auto-generated for no-merge-commit merge: !5183 merge master into master md大模型检测低错修改 Created-by: gitcode-chenjiao Commit-by: gitcode-chenjiao Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动，包括改动的原因和所采取的方法。--> md大模型检测低错修改 ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。--> <!-- 如果这个PR是为了解决特定的问题单，请在这里描述问题单单号。--> [# 2466](https://gitcode.com/cann/ops-math/issues/2446) ## 测试 <!--描述进行了哪些测试来验证你的改动。包括但不限于二级冒烟、算子泛化等。--> ok ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> *.md ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [x] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!5183	18 天前

Sub

产品支持情况

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

功能说明

算子功能：完成减法计算，被减数按alpha进行缩放。
计算公式：
$outi=selfi−alpha×otheriout_{i} = self_{i} - alpha \times other_{i}$ $selfRefi=selfRefi−alpha×otheriselfRef_{i}=selfRef_{i}-alpha \times other_{i}$

参数说明

参数名	输入/输出/属性	描述	数据类型	数据格式
self	输入	待进行sub计算的入参，公式中的self_i。	FLOAT、FLOAT16、DOUBLE、INT32、INT64、INT16、INT8、UINT8、BOOL、COMPLEX128、COMPLEX64、BFLOAT16	ND
other	输入	待进行sub计算的入参，公式中的other_i。	FLOAT、FLOAT16、DOUBLE、INT32、INT64、INT16、INT8、UINT8、BOOL、COMPLEX128、COMPLEX64、BFLOAT16	ND
alpha	输入	待进行sub计算的入参，公式中的alpha。	FLOAT、FLOAT16、DOUBLE、INT32、INT64、INT16、INT8、UINT8、BOOL、COMPLEX128、COMPLEX64、BFLOAT16	ND
out	输出	待进行sub计算的出参，公式中的out_i。	FLOAT、FLOAT16、DOUBLE、INT32、INT64、INT16、INT8、UINT8、BOOL、COMPLEX128、COMPLEX64、BFLOAT16	ND

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

约束说明

无

调用说明

调用方式	调用样例	说明
aclnn调用	test_aclnn_sub	通过aclnnSub接口方式调用Sub算子。
图模式调用	test_geir_sub	通过算子IR构图方式调用Sub算子。