Star0

Star0

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

c4693c96创建于 2 天前历史提交

文件	最后提交记录	最后更新时间
docs	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	3 天前
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	1 个月前
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	2 个月前
op_api	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	2 天前
op_graph	fix(proto): 修复39个算子proto.h注释的拼写与语法错误 Co-authored-by: sun-yibo1<sunyibo1@hisilicon.com> # message auto-generated for no-merge-commit merge: !5815 merge fix/proto-comment-grammar into master fix(proto): 修复39个算子proto.h注释的拼写与语法错误 Created-by: sun-yibo1 Commit-by: sun-yibo1 Merged-by: cann-robot Description: ## 描述对 39 个算子的 `op_graph/_proto.h` 注释做拼写与语法修复，仅注释与格式变更，不涉及任何代码逻辑： - 拼写错误：select_v2 `condittion`→`condition`（2处）、axpy_v2 `opratioan`→`operation`、clip_by_value 英文 dtype 列表中的全角逗号改半角 - 系统性语法（历史模板复制扩散）：`Support broadcasting`→`Supports broadcasting`（27个文件）、`A ND`→`An ND`（33个文件97处） - 单文件语法修复 71 处（含 signbit `returns`→`Returns`、axpy `should`→`Should` 两处句首大写补漏）*：ger `has`→`have`、addcmul/addcdiv `multiply`→`multiplies`、muls `An required`→`A required`、equal `(x=y)`→`(x1==x2)`、`Elementwise`→`Element-wise`、bias_add/lerp/gcd 语序与冠词修复、`aicore`→`AiCore`、mod/floor_mod 矛盾表述统一、句首大小写与标点空格等 - 附带 pre-commit 钩子强制规范化（clang-format 重排、尾随空白、文件末尾换行），并保留各文件原有行尾风格（CRLF/LF）涉及算子：add/sub/mul/muls/div/real_div/floor_div/div_no_nan/pow/minimum/maximum/signbit/less/less_equal/greater/greater_equal/not_equal/equal/bias_add/squared_difference/lerp/logical_and/logical_or/bitwise_or/bitwise_and/bitwise_xor/tanh_grad/axpy/axpy_v2/addcmul/addcdiv/select/select_v2/mod/floor_mod/ger/gcd/clip_by_value/clip_by_value_v2 ## 关联的Issue - #3234 - https://gitcode.com/cann/ops-math/issues/3234 ## 测试 - 纯注释变更，无功能影响 - pre-commit 钩子全部通过（clang-format / trailing-whitespace / end-of-file-fixer / codespell / OAT 合规检查） - 严格限定在上述 39 个算子文件，全仓其余算子未改动（其余算子同类问题可后续另行清理） ## 文档更新不涉及仓内 docs/ 目录；本 PR 即为 proto.h 注释（算子原型文档源）的修正 ## 类型标签 - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [x] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!5815	9 天前
op_host	feat(ascend960): 39个算子使能ascend960 Co-authored-by: hahaha22<wangrunze20@h-partners.com> # message auto-generated for no-merge-commit merge: !6137 merge feat/ascend960-enable into master feat(ascend960): 39个算子使能ascend960 Created-by: hahaha22 Commit-by: hahaha22 Merged-by: cann-robot Description: ## 描述为 ascend960 新增芯片形态适配算子，覆盖 math 与 conversion 目录下共 39 个算子，无 kernel 代码改动，复用现有 arch35 kernel 与 tiling。 ### 改动原因 ascend960 为新增芯片形态，现有算子需补齐平台注册、编译单元与二进制配置后，才能在新平台上被算子信息库发现、编译并选择正确的 kernel 二进制。 ### 改动方法 1. 算子信息库注册（`_def.cpp`）：新增 ascend960 平台注册（`AICore().AddConfig`） 2. 编译配置（`CMakeLists.txt`）：在 `SUPPORT_COMPUTE_UNIT` / `COMPUTE_UNITS` 中追加 ascend960，并在 `SUPPORT_TILING_DIR` 中追加对应的 `arch35`（tiling 目录沿用 arch35）；Add / ClipByValue / ClipByValueV2 的编译选项并入既有 `add_kernel_sources` 块（选项与 ascend950 / ascend350 既有块一致） 3. 二进制配置：为每个算子新增 ascend960 二进制配置目录，包含 `<op>_binary.json` 与 `<op>_simplified_key.ini`，内容沿用 ascend950 既有配置 4. 中央注册表（`scripts/kernel/binary_config/ascendc_config.json`）：33 个既有条目补齐 ascend960 编译选项，Mod 条目同步补齐（平台注册与编译单元主线已就绪） 5. 注册口径*：Signbit / Add / Ger / Gcd / BitwiseXor 不在中央注册表新增 ascend960 条目（Add 编译选项已迁移至 CMake 管理，其余沿主线现状，主线 ascend950 亦无对应条目） ### 算子清单（共 39 个） - math 目录 36 个：Add、Sub、Mul、Muls、Div、DivNoNan、RealDiv、FloorDiv、FloorMod、BitwiseAnd、BitwiseOr、BitwiseXor、Equal、NotEqual、Less、LessEqual、Greater、GreaterEqual、LogicalAnd、LogicalOr、Select、SelectV2、Maximum、Minimum、Pow、Lerp、BiasAdd、Axpy、AxpyV2、Addcmul、Addcdiv、Ger、Gcd、Signbit、TanhGrad、SquaredDifference、Mod - conversion 目录 2 个：ClipByValue、ClipByValueV2 ## 关联的Issue - https://gitcode.com/cann/ops-math/issues/3597 ## 测试 - pre-commit 全部 hook 通过（clang-format / OAT / codespell 等） - CI compile 门禁通过（18 个 Compile Job） - ascend960 内核编译与上板验证待具备 ascend960 环境后执行，合入前补充 ## 文档更新无（仅平台适配，不改变算子功能与约束，不涉及 README 更新） ## 类型标签 - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!6137	17 天前
op_kernel	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	21 天前
op_kernel_aicpu	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	21 天前
tests	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	2 天前
CMakeLists.txt	feat(ascend960): 39个算子使能ascend960 Co-authored-by: hahaha22<wangrunze20@h-partners.com> # message auto-generated for no-merge-commit merge: !6137 merge feat/ascend960-enable into master feat(ascend960): 39个算子使能ascend960 Created-by: hahaha22 Commit-by: hahaha22 Merged-by: cann-robot Description: ## 描述为 ascend960 新增芯片形态适配算子，覆盖 math 与 conversion 目录下共 39 个算子，无 kernel 代码改动，复用现有 arch35 kernel 与 tiling。 ### 改动原因 ascend960 为新增芯片形态，现有算子需补齐平台注册、编译单元与二进制配置后，才能在新平台上被算子信息库发现、编译并选择正确的 kernel 二进制。 ### 改动方法 1. 算子信息库注册（`_def.cpp`）：新增 ascend960 平台注册（`AICore().AddConfig`） 2. 编译配置（`CMakeLists.txt`）：在 `SUPPORT_COMPUTE_UNIT` / `COMPUTE_UNITS` 中追加 ascend960，并在 `SUPPORT_TILING_DIR` 中追加对应的 `arch35`（tiling 目录沿用 arch35）；Add / ClipByValue / ClipByValueV2 的编译选项并入既有 `add_kernel_sources` 块（选项与 ascend950 / ascend350 既有块一致） 3. 二进制配置：为每个算子新增 ascend960 二进制配置目录，包含 `<op>_binary.json` 与 `<op>_simplified_key.ini`，内容沿用 ascend950 既有配置 4. 中央注册表（`scripts/kernel/binary_config/ascendc_config.json`）：33 个既有条目补齐 ascend960 编译选项，Mod 条目同步补齐（平台注册与编译单元主线已就绪） 5. 注册口径*：Signbit / Add / Ger / Gcd / BitwiseXor 不在中央注册表新增 ascend960 条目（Add 编译选项已迁移至 CMake 管理，其余沿主线现状，主线 ascend950 亦无对应条目） ### 算子清单（共 39 个） - math 目录 36 个：Add、Sub、Mul、Muls、Div、DivNoNan、RealDiv、FloorDiv、FloorMod、BitwiseAnd、BitwiseOr、BitwiseXor、Equal、NotEqual、Less、LessEqual、Greater、GreaterEqual、LogicalAnd、LogicalOr、Select、SelectV2、Maximum、Minimum、Pow、Lerp、BiasAdd、Axpy、AxpyV2、Addcmul、Addcdiv、Ger、Gcd、Signbit、TanhGrad、SquaredDifference、Mod - conversion 目录 2 个：ClipByValue、ClipByValueV2 ## 关联的Issue - https://gitcode.com/cann/ops-math/issues/3597 ## 测试 - pre-commit 全部 hook 通过（clang-format / OAT / codespell 等） - CI compile 门禁通过（18 个 Compile Job） - ascend960 内核编译与上板验证待具备 ascend960 环境后执行，合入前补充 ## 文档更新无（仅平台适配，不改变算子功能与约束，不涉及 README 更新） ## 类型标签 - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!6137	17 天前
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	1 个月前

Sub

产品支持情况

产品	是否支持
Ascend 950PR&950DT系列产品	√
Atlas A3系列产品	√
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算子。