ops-math/math/sub/op_api · SeanDictionary/ops-math

cann-robotfix(pow/sub): A2/A3补充out dtype白名单校验（#3964 ）

Files	Last commit	Last update
aclnn_frac.cpp	refactor: 继续迁移非 experimental 公共头文件引用 Co-authored-by: clinglai0517<laichangling@huawei.com> # message auto-generated for no-merge-commit merge: !6220 merge refactor/math-prefix-common-headers-1008 into master refactor: 继续迁移非 experimental 公共头文件引用 Created-by: clinglai0517 Commit-by: clinglai0517 Merged-by: cann-robot Description: ## 描述继续将 ops-math 的公共 op_api 头文件引用迁移到带 `_math` 后缀的名称，减少多个算子仓库集成时同名头文件误引用的风险。本次更新 35 个非 experimental 文件中的 35 处引用： - `level2_base.h` → `level2_base_math.h`：11 处。 - `level2_base_caculation.h` → `level2_base_caculation_math.h`：1 处。 - `op_api_def.h` → `op_api_def_math.h`：23 处。对涉及文件同步执行仓库 clang-format、尾部空白及 EOF 规范化。此次共 35 个文件，新增 2506 行、删除 2419 行；除头文件引用替换外，其余为格式调整。 `sin` 因此前发现的 UT 全局平台状态污染继续暂缓；experimental 文件继续分批处理。现有兼容转发头保留。 ## 关联的Issue 延续公共头文件重名问题的分阶段迁移：https://gitcode.com/cann/ops-math/issues/3243 ## 测试在服务器 `192.168.9.234` 的 `lcl_ops_9.0` 容器（CANN 9.0.0 / aarch64）中，以基线 `16700ac8d829f8e66180c612f45ed4384b51e2e0` 创建独立工作区验证。35 个修改文件的 SHA256 与本地逐一一致。 - `git diff --check`：通过。 - pre-commit：所有 hook 通过；另显式扫描全部 35 文件的 OAT 检查通过。 - `bash build.sh --opapi -j16`：全量编译通过，生成 `libopapi_math.so`。 - 仓库 CI 脚本为涉及的 30 个算子生成 op_api 单测命令，指定 `ascend910b` 执行：605 个用例全部通过。 - 额外执行 `fill_diagonal_v2,flatten,slogdet,svd` 的 op_host 验证：容器中 `libophost_math_ut.so` 加载时缺少 `ops::OpAICoreDef::AddConfig(const char*, ops::OpAICoreConfig&)` 符号，导致注册失败及 FillDiagonalV2 InferShape 单测段错误。去掉全部 35 文件修改后的干净基线使用相同命令复现相同失败；该失败独立于本次头文件引用替换，op_host 验证尚未通过。 ## 文档更新不涉及用户文档更新。 ## 类型标签 - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [x] 其他，请描述：公共头文件引用重构及格式规范化。 See merge request: cann/ops-math!6220	27 days ago
aclnn_frac.h	fix(math): 修正equal/frac/affine_grid算子文档与头文件描述对齐实现 Co-authored-by: chenyifeng<chenyifeng27@h-partners.com> # message auto-generated for no-merge-commit merge: !6852 merge fix_MD_1029 into master fix(math): 修正equal/frac/affine_grid算子文档与头文件描述对齐实现 Created-by: chen_0715 Commit-by: chenyifeng Merged-by: cann-robot Description: ## 描述本 PR 汇总修正 4 处 math 算子文档/头文件注释与实现不一致的问题，均为描述性修正，不涉及代码逻辑改动： 1. aclnnIsInScalarTensor（math/equal/docs/aclnnIsInScalarTensor.md 及英文版）：out 参数说明由「shape与testElements相同」修正为「0维标量」，维度列改为 0，并同步修正错误码表 161002 子项为「testElements的维度超过8维」「out的维度不是0维」，对齐实现 OP_CHECK_WRONG_DIMENSION(out, 0)，消除文档自相矛盾。 2. aclnnInplaceFrac（math/sub/docs/aclnnFrac&aclnnInplaceFrac.md）：inputRef 参数角色由「输入」修正为「输入/输出」，补充「计算结果原地写回该tensor」语义说明。 3. aclnnAffineGrid（math/affine_grid/docs/aclnnAffineGrid.md）：错误码表 161002 新增「theta为空tensor」子项（对齐实现空 tensor 校验）；约束说明补充「size中的C不参与计算，对取值无约束」。 4. aclnnFrac（math/sub/op_api/aclnn_frac.h）：头文件注释三处 dtype 声明补齐 BFLOAT16，并注明「在Atlas训练系列产品上不支持该数据类型」，对齐资料与实现（A2/A3/950 支持 BF16，Atlas 训练系列不支持）。 ## 关联的Issue - #4109（aclnnIsInScalarTensor out shape 描述与实现矛盾） - #4134（aclnn_frac.h 头文件 dtype 声明缺 BFLOAT16） - [#4124](https://gitcode.com/cann/ops-math/issues/4124) - [#4129](https://gitcode.com/cann/ops-math/issues/4129) ## 测试本次改动均为文档/注释修正，不涉及代码逻辑，无运行测试需求。相关行为已由实现侧真机验证：aclnnIsInScalarTensor 在 A2 910B2 上 out 为 0 维标量正常返回（161002 仅在 out 非 0 维时上报）；aclnnFrac 的 BFLOAT16 在 A2 上 407 条用例全部通过。 ## 文档更新 - math/equal/docs/aclnnIsInScalarTensor.md - math/equal/docs/aclnnIsInScalarTensor_en.md - math/sub/docs/aclnnFrac&aclnnInplaceFrac.md - math/affine_grid/docs/aclnnAffineGrid.md - math/sub/op_api/aclnn_frac.h（doxygen 注释） ## 类型标签 - [x] Bug修复 - [x] 文档更新 - [ ] 新特性 - [ ] 性能优化 - [ ] 其他，请描述： See merge request: cann/ops-math!6852	11 days ago
aclnn_rsub.cpp	fix: add/sub/rsub的IsEqualToOne在alpha为NaN时误判为1（isnan守卫式，!6150备选方案） Co-authored-by: hahaha22<wangrunze20@h-partners.com> # message auto-generated for no-merge-commit merge: !6154 merge fix/add-sub-alpha-nan-isnan into master fix: add/sub/rsub的IsEqualToOne在alpha为NaN时误判为1（isnan守卫式，!6150备选方案） Created-by: hahaha22 Commit-by: hahaha22 Merged-by: cann-robot Description: 修复 `IsEqualToOne` 的 RegBase 分支在 alpha 为 NaN 时误判为 1 的缺陷。缺陷 issue：#3460 ## 问题与根因 `IsEqualToOne` 的 RegBase 分支使用 `!(x > 1 \|\| x < 1)` 判断 alpha 是否为 1。IEEE-754 下 NaN 与任何数比较均为 false，`NaN > 1` 与 `NaN < 1` 同时为 false，取反后整体为 true —— NaN alpha 被误判为 1，`aclnnAdd` / `aclnnSub` / `aclnnRsub` 跳过 alpha 乘法直接走快速路径，输出 `self ± other` 而非 NaN（竞品 PyTorch 为全 NaN）。 ## 修复（isnan 守卫式）不替换既有判断式结构，在其前叠加显式 NaN 排除，取值缓存到局部变量避免重复调用访问器： `cpp if (calcType == DataType::DT_DOUBLE) { double alphaVal = alpha->ToDouble(); return !std::isnan(alphaVal) && !(alphaVal > 1 \|\| alphaVal < 1); } float alphaVal = alpha->ToFloat(); return !std::isnan(alphaVal) && !(alphaVal > 1 \|\| alphaVal < 1);` NaN 显式排除后按原判断式求值，不再误判为 1，NaN 按语义传播。 ## 修改范围 - `math/add/op_api/aclnn_add.cpp`、`math/sub/op_api/aclnn_sub.cpp`、`math/sub/op_api/aclnn_rsub.cpp` 各 `IsEqualToOne` 的 DT_DOUBLE 与默认分支 - 仅 RegBase 路径；`!IsRegBase()` 早退分支（非 RegBase 平台行为）保持不变 ## 自测 - 谓词级对照：修复前后在 14 组 float 边界取值（1±1ulp、±inf、denormal、±0 等）上行为不变，仅 NaN 由 true（缺陷）变为 false - `g++ -fsyntax-only`（CANN 9.2.0 真实构建参数，含 `-Werror`）：3 文件 0 error（同时验证 `std::isnan` 经既有传递包含可解析，与文件内既有 `std::abs` 用法同源） - `clang-format --style=file --dry-run -Werror`：3 文件 0 告警 See merge request: cann/ops-math!6154	1 month ago
aclnn_rsub.h	refactor: 统一 add/mul/pow/real_div/sub 算子代码风格 Co-authored-by: cai-chengchao<caichengchao1@h-partners.com> # message auto-generated for no-merge-commit merge: !6053 merge ccc_0929 into master refactor: 统一 add/mul/pow/real_div/sub 算子代码风格 Created-by: cai-chengchao Commit-by: cai-chengchao Merged-by: cann-robot Description: ## 描述对 add、mul、pow、real_div、sub 五个数学算子的代码进行统一的代码风格规范化（clang-format），包括： - 统一命名空间/类的括号风格与缩进 - 统一指针、引用修饰符位置（如 `const char* const`） - 规范函数声明与调用的换行对齐 - 统一注释风格及文件结尾换行符 - 修正 real_div 配置注释中的拼写错误（build-in → built-in）本次改动仅涉及代码格式调整，不包含任何功能或逻辑变更。 ## 关联的Issue - #3464 ## 测试无功能变更，仅代码格式调整，未新增测试。 ## 文档更新无。 ## 类型标签 - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [x] 其他，请描述：代码风格规范化 See merge request: cann/ops-math!6053	29 days ago
aclnn_sub.cpp	fix(pow/sub): A2/A3补充out dtype白名单校验（#3964） Co-authored-by: doufloat<baijinxiang@huawei.com> # message auto-generated for no-merge-commit merge: !6800 merge fix/pow-out-dtype-whitelist into master fix(pow/sub): A2/A3补充out dtype白名单校验（#3964） Created-by: doufloat Commit-by: doufloat Merged-by: cann-robot Description: ## 描述问题 `aclnnPowScalarTensor`、`aclnnPowTensorScalar`、`aclnnPowTensorTensor`、`aclnnSubs` 和 `aclnnSub` 在 `out` 传入资料输出支持列表之外的 dtype（如 `DT_INT4`）时，A2/A3 会继续进入 Build/Cast 并返回内部错误码，而不是第一段接口约定的 `161002`。根因输出类型此前只经过可转换关系检查。部分非法 dtype 能被 `CanCast` 放行，随后在 Build/Cast 阶段失败，导致参数非法被降级为内部错误。方案 - 新增 out dtype 白名单校验，但仅在 `NpuArch::DAV_2201`（A2/A3）启用。 - `aclnn_pow.cpp` 和 `aclnn_pow_tensor_tensor.cpp` 覆盖 `aclnnPowScalarTensor`、`aclnnPowTensorScalar`、`aclnnPowTensorTensor` 三个 out-of-place 入口。 - `aclnn_sub.cpp` 在 `CheckParamsScalar` 和 `CheckParams` 中覆盖 `aclnnSubs`、`aclnnSub`；原地接口继续沿用原有输入别名校验。 - A5 走 `DAV_3510` 路径，不启用新增白名单，保持原有 `CanCast` 及后续 Build 行为不变。 - A2/A3 合法 out dtype 路径不变，仅将非法 dtype 的返回码修正为 `ACLNN_ERR_PARAM_INVALID(161002)`。 ## 关联的Issue https://gitcode.com/cann/ops-math/issues/3964 ## 测试 A2：Ascend 910B3，CANN 9.2.0 - `math_op_api_ut`：`158/158 PASS`，包含新增 A2 非法 out 校验和 A5 不拦截回归用例。 - 当前补丁真机矩阵：`PowScalarTensor/PowTensorScalar/PowTensorTensor/Subs/Sub` 的 `INT4/UINT1/HIFLOAT8/FLOAT8_E5M2/COMPLEX32` 均返回 `161002`。 - 合法路径未被误拦截：三个 Pow 入口的 `INT64 -> UINT16/UINT32/UINT64` 均返回 `0`；`FLOAT/FLOAT16/DOUBLE/BF16/COMPLEX64/COMPLEX128` 输出在对应可转换输入下保持 `0`。 A5：Ascend 950PR，CANN 9.2.0 - `math_op_api_ut`：`158/158 PASS`，包含 `Ascend950_out_dtype_check_not_applied` 回归用例。 - 当前补丁真机矩阵与基线一致： \| out dtype \| PowScalarTensor \| PowTensorScalar / PowTensorTensor / Subs / Sub \| \| --- \| --- \| --- \| \| INT4 / UINT1 / COMPLEX32 \| 561000 \| 561103 \| \| HIFLOAT8 / FLOAT8_E5M2 \| 0 \| 0 \| - 新增 A2/A3 out 白名单在 A5 未生效；合法 `INT64 -> UINT16/UINT32/UINT64` 仍返回 `0`。实现路径说明 A3 与 A2 均走 `DAV_2201` 路径，共用同一实现。本轮完成 A2 和 A5 真机验证。 ## 文档更新不涉及。 ## 类型标签 - [x] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他（请说明） See merge request: cann/ops-math!6800	10 days ago
aclnn_sub.h	refactor: 统一 add/mul/pow/real_div/sub 算子代码风格 Co-authored-by: cai-chengchao<caichengchao1@h-partners.com> # message auto-generated for no-merge-commit merge: !6053 merge ccc_0929 into master refactor: 统一 add/mul/pow/real_div/sub 算子代码风格 Created-by: cai-chengchao Commit-by: cai-chengchao Merged-by: cann-robot Description: ## 描述对 add、mul、pow、real_div、sub 五个数学算子的代码进行统一的代码风格规范化（clang-format），包括： - 统一命名空间/类的括号风格与缩进 - 统一指针、引用修饰符位置（如 `const char* const`） - 规范函数声明与调用的换行对齐 - 统一注释风格及文件结尾换行符 - 修正 real_div 配置注释中的拼写错误（build-in → built-in）本次改动仅涉及代码格式调整，不包含任何功能或逻辑变更。 ## 关联的Issue - #3464 ## 测试无功能变更，仅代码格式调整，未新增测试。 ## 文档更新无。 ## 类型标签 - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [x] 其他，请描述：代码风格规范化 See merge request: cann/ops-math!6053	29 days ago
sub.cpp	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	1 month ago
sub.h	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	10 months ago