dc6d8a58创建于 10 天前历史提交

文件	最后提交记录	最后更新时间
docs	math仓算子资料修改 Co-authored-by: FelixTang7<tangyanfeng@huawei.com> # message auto-generated for no-merge-commit merge: !5405 merge md into master math仓算子资料修改 Created-by: FelixTang7 Commit-by: FelixTang7 Merged-by: cann-robot Description: ## 描述 math仓部分资料不清晰，需要进行修改 ## 关联的Issue https://gitcode.com/cann/ops-math/issues/3040 ## 测试 <!--描述进行了哪些测试来验证你的改动。包括但不限于二级冒烟、算子泛化等。--> ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [x] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!5405	14 天前
examples	算子资料修改后同步修改example Co-authored-by: FelixTang7<tangyanfeng@huawei.com> # message auto-generated for no-merge-commit merge: !5542 merge md into master 算子资料修改后同步修改example Created-by: FelixTang7 Commit-by: FelixTang7 Merged-by: cann-robot Description: ## 描述部分算子资料有误，修改算子文档后同步对example进行修改 ## 关联的Issue https://gitcode.com/cann/ops-math/issues/3040 ## 测试文档修改，无需测试 ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [x] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!5542	10 天前
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	10 天前
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	feat: mul/select 算子支持 PcieThrough 场景识别与回退 Co-authored-by: hahaha22<wangrunze20@h-partners.com> # message auto-generated for no-merge-commit merge: !5116 merge pcie_through into master feat: mul/select 算子支持 PcieThrough 场景识别与回退 Created-by: hahaha22 Commit-by: hahaha22 Merged-by: cann-robot Description: ## 描述 mul、select 算子增加 PcieThrough（PCIe 直通）场景适配： 1. select simt tiling 回退：`SelectSimtTiling::IsCapable` 新增前置 `IsPcieThrough()` 判断，PcieThrough 场景返回 false，回退到低优先级的 `SelectTiling`（内部走 opbase broadcast 模板，PcieThrough 下强制 UB BRC 调度） 2. `IsPcieThrough` 实现与 broadcast 模板版本兼容方案对齐：高版本 SDK（METADEF_VERSION_NUM >= 90200000）调用 `context_->GetPcieThroughFlag()`，低版本走 `#else` 兜底 false 3. PcieThrough 能力标记：`IMPL_OP_INFERSHAPE(Mul)` 与 `IMPL_OP_INFERSHAPE(Select)` 在高版本 SDK 下链式调用 `SetSupportPcieThrough()` 4. 显式包含 `version/metadef_version.h` 5. 顺带对 select_simt_tiling.cpp/.h 与 mul_infershape.cpp 做格式化规整（缩进、大括号风格、文件末尾换行），无逻辑变更配套模板侧 PR：https://gitcode.com/cann/opbase/pull/788（broadcast 模板 PcieThrough 识别与 UB BRC 强制调度）。本 PR 依赖其先合入，否则 PcieThrough 场景下 broadcast 模板缺少强制 UB BRC 与非连续拦截，行为不完整。 ## 关联的Issue - https://gitcode.com/cann/ops-math/issues/3018 ## 测试 - 当前 SDK metadef 版本低于 90200000，`GetPcieThroughFlag`/`SetSupportPcieThrough` 真实分支未编入，兜底路径下行为与合入前一致 - `IsCapable` 回退分支与 `SetSupportPcieThrough` 标记待 METADEF >= 90200000 环境编译运行验证 - CI 流水线运行中（ci-pipeline-running） ## 文档更新不涉及 ## 类型标签 - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!5116	16 天前
op_kernel	refactor: convert math SIMT kernels to C-style APIs Co-authored-by: qianzehong<qianzehong@huawei.com> # message auto-generated for no-merge-commit merge: !2616 merge c_style into master refactor: convert math SIMT kernels to C-style APIs Created-by: qianzehong Commit-by: qianzehong Merged-by: cann-robot Description: ## 描述对 ops-math 中 `floor_div`、`floor_mod`、`mul`、`select`、`mod`、`gcd` 算子的 Ascend 950 arch35 kernel 做 SIMT C 风格 API 整改。主要变更： - 增加 `simt_api/asc_simt.h` 头文件依赖。 - 将线程索引/线程数 API 从 `Simt::GetThreadIdx/GetThreadNum` 替换为 `threadIdx.x/blockDim.x`。 - 将 SIMT 调用 API 从 `Simt::VF_CALL/Dim3` 替换为 `asc_vf_call/dim3`。 - `select` 中 `Simt::UintDiv` 因当前 CANN 9.0.0 SIMT 头文件无 C 风格等价 API，保持不变。 ## 关联的Issue https://gitcode.com/cann/ops-math/issues/1481 ## 测试按本次处理要求跳过编译验证；本地 CANN 包版本与目标验证环境不一致。 ## 文档更新无文档更新。 ## 类型标签 - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [x] 其他，请描述：SIMT C 风格 API 整改 See merge request: cann/ops-math!2616	4 个月前
op_kernel_aicpu	feat: Equal/InvGrad/Mul/RealDiv/ReduceMean/RightShift 使能 AICPU 常量折叠 Co-authored-by: jialimin1<jialimin1@huawei.com> # message auto-generated for no-merge-commit merge: !4965 merge enable-aicpu-const-folding into master feat: Equal/InvGrad/Mul/RealDiv/ReduceMean/RightShift 使能 AICPU 常量折叠 Created-by: jialimin1 Commit-by: jialimin1 Merged-by: cann-robot Description: ## 描述为 Equal、InvGrad、Mul、RealDiv、ReduceMean 和 RightShift 使能 AICPU 常量折叠： 1. 在 6 个算子的 `CMakeLists.txt` 中增加 `HOSTCPU TRUE`，将 AICPU kernel 纳入 HostCPU 构建。 2. 引入 `aicpu/math_aicpu_register.h`，将原有 `REGISTER_CPU_KERNEL` 注册切换为 `OPS_MATH_REGISTER_CPU_KERNELV2`。 3. 保持现有计算实现、数据类型范围和硬件支持范围不变；RealDiv 继续同时注册 `RealDiv` 与 `Div`。 ## 关联的Issue 关联Issue #2807 ## 测试建议按以下场景验证本次变更： 1. 编译构建测试：执行 CMake 配置并编译 6 个算子，确认 `HOSTCPU TRUE` 配置及 V2 注册宏编译、链接正常。 2. 注册加载验证：确认 Equal、InvGrad、Mul、RealDiv、Div、ReduceMean 和 RightShift 的 HostCPU kernel 均可被正确发现和加载。 3. 常量折叠功能测试：为每个算子构造常量输入，确认编译阶段生成的折叠结果与直接执行 AICPU kernel 的结果一致。 4. 数据类型与 shape 覆盖：覆盖各算子已有的代表性数据类型、标量/张量输入、广播或属性组合。 5. 异常场景测试：验证非法数据类型、shape 或属性仍按原有逻辑返回错误。 6. 回归测试：确认非恒定输入仍走原有运行时执行路径，且不同 `SUPPORT_COMPUTE_UNIT` 下的构建行为不变。 > 以上为根据代码变更推导的测试场景，未代替实际测试结果；请由提交者补充二级冒烟、算子泛化及执行结果。 ## 文档更新无。此次仅调整构建配置和 AICPU kernel 注册方式，不涉及算子用户文档或接口变更。 ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!4965	27 天前
tests	新增Mul算子AICPU实现 Co-authored-by: liu-wei<lovline.liuwei@huawei.com> # message auto-generated for no-merge-commit merge: !4111 merge master_mul_aicpu into master 新增Mul算子AICPU实现 Created-by: liu-wei Commit-by: liu-wei Merged-by: cann-robot Description: Closes #2284 ## 描述为 Mul 算子新增 AICPU 后端实现，支持 AI CPU 上执行逐元素乘法计算。 ### 新增文件 \| 文件 \| 行数 \| 说明 \| \|------\|------\|------\| \| `mul_aicpu.h` \| 46 \| CpuKernel 类声明，含模板方法分发与对齐检查 \| \| `mul_aicpu.cpp` \| 487 \| 基于 Eigen 的 Mul 内核实现，支持广播和多数据类型 \| \| `mul_aicpu_def.cpp` \| 33 \| OpDef 注册，支持 13 种数据类型 \| \| `test_geir_mul.cpp` \| 309 \| GE IR 测试示例 \| \| `test_mul_aicpu.cpp` \| 267 \| 单元测试 \| ### 修改文件 - `docs/zh/op_list.md`：Mul 支持后端更新为 AI Core/AI CPU ### 关键特性 - 支持 INT8/UINT8/INT16/UINT16/INT32/UINT64/FLOAT/FLOAT16/BF16/DOUBLE/COMPLEX64/COMPLEX128 共 13 种数据类型 - 支持广播（broadcast）语义 - 使用 Eigen Tensor 库实现高性能计算 - 自动并行处理阈值：同 shape 7K / 中等 33K / 通用 6K ### 测试情况 - [ ] 单元测试通过 - [ ] 算子泛化验证 ### 类型标签 - [x] 新特性 🤖 Generated with [Claude Code](https://claude.com/claude-code) See merge request: cann/ops-math!4111	1 个月前
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	20 天前
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	23 天前

Mul

产品支持情况

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

功能说明

算子功能：实现两输入tensor逐元素相乘。
计算公式：

x_1 \times x_2

参数说明

参数名	输入/输出/属性	描述	数据类型	数据格式
x1	输入	待进行Mul计算的入参，公式中的x1。	FLOAT、FLOAT16、BFLOAT16、DOUBLE、INT8、UINT8、INT16、INT32、INT64、COMPLEX32、COMPLEX64、BOOL	ND
x2	输入	待进行Mul计算的入参，公式中的x2。	FLOAT、FLOAT16、BFLOAT16、DOUBLE、INT8、UINT8、INT16、INT32、INT64、COMPLEX32、COMPLEX64、BOOL	ND
y	输出	tensor逐元素乘积结果，公式中的y。	FLOAT、FLOAT16、BFLOAT16、DOUBLE、INT8、UINT8、INT16、INT32、INT64、COMPLEX32、COMPLEX64、BOOL	ND

约束说明

无

调用说明

调用方式	调用样例	说明
aclnn调用	test_aclnn_mul	通过aclnnMul接口方式调用Mul算子。