d679322b创建于 12 天前历史提交

文件	最后提交记录	最后更新时间
docs	产品名变更 Co-authored-by: gitcode-chenjiao<chenjiao31@huawei.com> # message auto-generated for no-merge-commit merge: !5912 merge master into master 产品名变更 Created-by: gitcode-chenjiao Commit-by: gitcode-chenjiao Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动，包括改动的原因和所采取的方法。--> AI处理器对应的产品型号名称更新 ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。--> <!-- 如果这个PR是为了解决特定的问题单，请在这里描述问题单单号。--> [#3258](https://gitcode.com/cann/ops-math/issues/3258) ## 测试 <!--描述进行了哪些测试来验证你的改动。包括但不限于二级冒烟、算子泛化等。--> ok ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> *.md ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [x] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!5912	12 天前
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	22 天前
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	22 天前
op_graph	add range mod adj pow one_hot Co-authored-by: Nicooo-Wang<wangtianle3@huawei.com> # message auto-generated for no-merge-commit merge: !541 merge move-ops into master add range mod adj pow one_hot Created-by: Nicooo-Wang Commit-by: Nicooo-Wang;王天乐 Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动，包括改动的原因和所采取的方法。--> ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。--> <!-- 如果这个PR是为了解决特定的问题单，请在这里描述问题单单号。--> ## 测试 <!--描述进行了哪些测试来验证你的改动。包括但不限于二级冒烟、算子泛化等。--> ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!541	8 个月前
op_host	【mc62】workspaceSize硬编码16M大小，替换为通过接口获取，ascendcPlatform.GetLibApiWorkSpaceSize(); Co-authored-by: xufeng12121<1074805447@qq.com> # message auto-generated for no-merge-commit merge: !5675 merge math_space into master 【mc62】workspaceSize硬编码16M大小，替换为通过接口获取，ascendcPlatform.GetLibApiWorkSpaceSize(); Created-by: xufeng12121 Commit-by: xufeng12121 Merged-by: cann-robot Description: ## 描述 1. 背景：mc62算子workspace占用对比1951增大 2. 由于mc62代码和A5是onetrack, 所有workspacesize硬编码为16M, 在mc62上也同样是16M，asc 提供接口，对mc62进行区分，从原来的16M 变成1K，A5和A2大小不变。 3. 修改：算子侧tiling中硬编码为16M大小的workspacesize统一改为用接口获取。 GetLibApiWorkSpaceSize实现如下： const static uint32_t WORK_SPACE_SIZE_910B = 16 * 1024 * 1024; const static uint32_t WORK_SPACE_SIZE_950 = 16 * 1024 * 1024; const static uint32_t WORK_SPACE_SIZE = 1024; uint32_t PlatformAscendC::GetLibApiWorkSpaceSize(void) const { auto npuArch = GetCurNpuArch(); if (npuArch == NpuArch::DAV_RESV) { PF_LOGE("get platform failed, CurNpuArch is NpuArch::DAV_RESV"); return -1; } else if (npuArch == NpuArch::DAV_2201) { return WORK_SPACE_SIZE_910B; } else if (npuArch == NpuArch::DAV_3510) { return WORK_SPACE_SIZE_950; } return WORK_SPACE_SIZE; } ## 关联的Issue https://gitcode.com/cann/ops-math/issues/3232 ## 测试 1.单算子框架测试通过 2.二级冒烟通过文档更新类型标签 - [x] Bug修复 See merge request: cann/ops-math!5675	15 天前
op_kernel	refactor: arch35算子Reg级API重命名迁移 Co-authored-by: hahaha22<wangrunze20@h-partners.com> # message auto-generated for no-merge-commit merge: !4468 merge work_8 into master refactor: arch35算子Reg级API重命名迁移 Created-by: hahaha22 Commit-by: hahaha22 Merged-by: cann-robot Description: ## 描述将 arch35 架构下 40 个算子 kernel 文件中的 Reg 级 API 名称从旧命名迁移到新命名，与已迁移的算子（xlogy、dawsn、rint、cdist、reduce_all、power 等）保持一致。涉及的 API 重命名映射： - `__local_mem__` → `__ubuf__`（地址空间限定符统一） - `Reg::DataCopy` (LoadDist) → `Reg::LoadAlign`（UB 到寄存器的对齐加载） - `Reg::DataCopy` (StoreDist) → `Reg::StoreAlign`（寄存器到 UB 的对齐存储） - `Reg::DataCopyUnAlignPre` → `Reg::LoadUnAlignPre` - `Reg::DataCopyUnAlign` → `Reg::LoadUnAlign` - `Reg::MaskPack` → `Reg::Pack` - `Reg::MaskUnPack` → `Reg::UnPack` - `Reg::MaskAnd` → `Reg::And`（与已有 Reg::And 通过重载统一） - `Reg::CompareScalar` → `Reg::Compares` - `Reg::UnalignReg` → `Reg::UnalignRegForLoad` 涉及的算子包括：clip_by_value、clip_by_value_v2、adjacent_difference、bitwise_xor、div、floor_div、floor_mod、lerp、logical_or、mod、pow、signbit、tanh_grad。此外，adjacent_difference 算子中存在原有的 `__local_mem__`/`__ubuf__` 地址空间限定符不一致问题（左侧声明与右侧转换不匹配），本次重命名一并修复。 ### 改动原因 AscendC SDK 更新了 Reg 级 API 命名规范，将原有的 `DataCopy` 拆分为语义更明确的 `LoadAlign`/`StoreAlign`，并移除了 Mask 系列 API 的 "Mask" 前缀（通过重载统一）。本 PR 将相关算子迁移至新命名。 ### 改动方法纯机械重命名，无逻辑变更。所有改动均为 1:1 的 API 名称替换，增删行数完全对称（+417 -417）。Load/Store 语义区分基于原有模板参数（LoadDist → LoadAlign, StoreDist → StoreAlign）和参数顺序（寄存器在前为 Load，地址在前为 Store）。 ## 关联的Issue - #2529 ## 测试待补充：二级冒烟测试、算子泛化测试。 ## 文档更新无 ## 类型标签 - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [x] 其他，请描述：API 重命名重构 See merge request: cann/ops-math!4468	1 个月前
op_kernel_aicpu	新增AICPU pow算子 Co-authored-by: AlleyWang<wanglei391@huawei.com> # message auto-generated for no-merge-commit merge: !2575 merge master into master 新增AICPU pow算子 Created-by: AlleyWang Commit-by: AlleyWang Merged-by: cann-robot Description: ## 描述新增算子的AI CPU kernel实现，扩展CANN算子库在AI CPU上的算子支持范围。 ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。--> <!-- 如果这个PR是为了解决特定的问题单，请在这里描述问题单单号。--> ## 测试根据代码变更，测试场景如下：算子构建测试执行 cmake 配置和编译，验证 CMake 配置变更正确算子加载测试验证pow.json 接口注册成功单元测试 pow UT: bash build.sh -u --opkernel_aicpu --ops=pow --soc=ascend910b GE IR验证 ## 文档更新 docs/zh/op_list.md新增pow aicpu算子支持说明 math/pow/README.md增加pow算子的图模式调用说明 ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!2575	4 个月前
tests	新增 muls/not_equal/pow/range/real_div 的 golden.py Co-authored-by: 鲜磊<xianlei4@h-partners.com> # message auto-generated for no-merge-commit merge: !3265 merge g4 into master 新增 muls/not_equal/pow/range/real_div 的 golden.py 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!3265	3 个月前
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	1 个月前
README.md	产品名变更 Co-authored-by: gitcode-chenjiao<chenjiao31@huawei.com> # message auto-generated for no-merge-commit merge: !5912 merge master into master 产品名变更 Created-by: gitcode-chenjiao Commit-by: gitcode-chenjiao Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动，包括改动的原因和所采取的方法。--> AI处理器对应的产品型号名称更新 ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。--> <!-- 如果这个PR是为了解决特定的问题单，请在这里描述问题单单号。--> [#3258](https://gitcode.com/cann/ops-math/issues/3258) ## 测试 <!--描述进行了哪些测试来验证你的改动。包括但不限于二级冒烟、算子泛化等。--> ok ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> *.md ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [x] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!5912	12 天前

Pow

产品支持情况

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

功能说明

接口功能：exponent每个元素作为input对应元素的幂完成计算。
计算公式：

out_i = input_i^{exponent_i}

参数说明

参数名	输入/输出	描述	数据类型	数据格式
input	输入	表示底数。数据类型与exponent满足TensorScalar互推导关系。	FLOAT、FLOAT16、DOUBLE、INT16、INT32、INT64、INT8、UINT8、COMPLEX64、COMPLEX128、BFLOAT16	ND
exponent	输入	表示指数。数据类型与input满足TensorScalar互推导关系。	FLOAT、FLOAT16、DOUBLE、INT16、INT32、INT64、INT8、UINT8、COMPLEX64、COMPLEX128、BFLOAT16	ND
out	输出	表示input的exponent次幂。数据类型需要是input的数据类型与exponent的数据类型推导之后可转换的数据类型（参见互转换关系）。	FLOAT、FLOAT16、DOUBLE、INT16、INT32、INT64、INT8、UINT8、COMPLEX64、COMPLEX128、BFLOAT16	ND

约束说明

算子约束：INT32整型计算在如下范围以外的场景，会出现超时。

shape	exponent_value
<=100000（十万）	-200000000~200000000(两亿)
<=1000000（百万）	-20000000~20000000(两千万)
<=10000000（千万）	-2000000~2000000(两百万)
<=100000000（亿）	-200000~200000(二十万)
<=1000000000（十亿）	-20000~20000(两万)

调用说明

调用方式	样例代码	说明
aclnn调用	test_aclnn_exp2	通过aclnnExp2接口方式调用Pow算子。
aclnn调用	test_aclnn_pow_tensor_tensor	通过aclnnPowTensorTensor接口方式调用Pow算子。
图模式调用	test_geir_pow	通过算子IR构图方式调用Pow算子。