Star0

Star0

cann-robotfeat: vector算子批量适配ascend960pr(DV122)平台

4dc1d6b5Add on 3 days agoHistory

Files	Last commit	Last update
docs	master翻译合入 Co-authored-by: caiwenwen<caiwenwen6@h-partners.com> # message auto-generated for no-merge-commit merge: !6751 merge master into master master翻译合入 Created-by: caiwenwen Commit-by: caiwenwen Merged-by: cann-robot Description: ## 描述合入math仓翻译的内容 ## 关联的Issue https://gitcode.com/cann/ops-math/issues/4170 ## 测试 <!--描述进行了哪些测试来验证你的改动。包括但不限于二级冒烟、算子泛化等。--> ## 文档更新 docs/context，aclnn等商发文档 ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [x] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!6751	17 days ago
examples	math仓算子错误日志整改 Co-authored-by: qq_52874943<gaomingyuan3@h-partners.com> # message auto-generated for no-merge-commit merge: !5345 merge master into master math仓算子错误日志整改 Created-by: qq_52874943 Commit-by: qq_52874943 Merged-by: cann-robot Description: ## 描述本 PR 针对 math 仓算子日志进行整改（关联 Issue #2853），主要目的是规范算子错误日志的级别与内容。改动覆盖约 35 个文件，核心内容包括三类：将 examples/test_geir_*.cpp 示例程序中 GE 初始化失败、加图失败、运行图失败、反初始化失败等错误场景的日志级别由 INFO 修正为 ERROR；修正 op_kernel_aicpu、op_api、tiling、framework 插件中参数校验及错误提示日志的英文语法与拼写；同时清理行尾多余空白字符。整体为纯日志与文档性整改，不涉及算子计算逻辑与对外接口行为变更。 ## 关联的Issue 关联Issue[#2853](https://gitcode.com/cann/ops-math/issues/2853) ## 测试本地已完成新精度对比与性能测试，通过静态检查、蓝区冒烟、算子ST以及算子二级冒烟。 ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [x] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!5345	2 months ago
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	3 months ago
op_api	style(math): 修复 op_api 层 CodeCheck 静态检查规范问题 Co-authored-by: xiafengkou<luojiangtao4@huawei.com> # message auto-generated for no-merge-commit merge: !6425 merge fix/math-codecheck-opapi into master style(math): 修复 op_api 层 CodeCheck 静态检查规范问题 Created-by: xiafengkou Commit-by: xiafengkou Merged-by: cann-robot Description: ## 变更说明修复 CodeCheck 静态检查在 math 仓 op_api 层报告的规范问题，仅做等价修改，不改变任何业务逻辑与对外接口（aclnn API 签名/ABI 不变）。涉及规则与修改： 1. G.FUN.02（声明与定义参数名/限定不一致） - `math/sqrt/op_api/aclnn_sqrt.h/.cpp`：头文件两处声明形参 `opExecutor` 对齐为 `executor`，@param 注释同步更新（DFX 宏内部硬编码标识符 `executor`，故改声明侧而非定义侧，不影响 ABI/调用方）；移除 `aclnnSqrt`/`aclnnInplaceSqrt` 中 `const aclrtStream stream` 的多余顶层 const，与头文件声明保持一致。 - `math/arg_min/op_api/aclnn_argmin.cpp`：`aclnnArgMin` 定义移除 `const aclrtStream stream` 的多余 const，与头文件声明一致。 2. G.INC.12（内部链接函数缺少 static） - `math/greater_equal/op_api/aclnn_ge_scalar.cpp`：仅本文件内调用的 `aclnnGeScalarCommon` 增加 static，限定内部链接。 3. G.INC.02（冗余 include） - 删除 cumprod、cosh、sign_bits_pack、polar、grouped_bias_add_grad 五个 op_api 源文件中未使用的 `#include "op_api/op_api_def_math.h"`。 4. G.CNS.04（只读指针参数应加 const） - `math/cdist_grad/op_api/aclnn_cdist_backward.cpp`：`GetOutTensorWithValueZero` 形参加 const。 - `math/sqrt/op_api/aclnn_sqrt.cpp`、`math/log/op_api/aclnn_log10.cpp`、`math/cos/op_api/aclnn_cos.cpp`：`CheckInplaceDtypeValid` 形参加 const。 ## 关联 Issue ## 测试报告 - 编译验证（ascend950）： - `bash build.sh --soc=ascend950 --ops=sqrt,arg_min,greater_equal,cdist_grad,cos,cosh,cumprod,grouped_bias_add_grad,log,polar,sign_bits_pack -j16`：ophost + binary 编译成功。 - `bash build.sh --opapi --soc=ascend950 -j16`：opapi 库编译成功。 - opapi UT：`bash build.sh -u --opapi --soc=ascend950 -j16`，均通过。 ## 自测考虑 - [x] 编译通过 - [x] UT 验证通过 - [x] 保持原有逻辑与接口不变 ## 文档无 See merge request: cann/ops-math!6425	28 days ago
op_graph	feat: Infershape整改，25个math算子InferDataType注册迁移至op_graph目录（IMPL_OP独立链）及UT看护 Co-authored-by: ywx1080114<yuanbin22@huawei.com> # message auto-generated for no-merge-commit merge: !6392 merge infershape-dtype into master feat: Infershape整改，25个math算子InferDataType注册迁移至op_graph目录（IMPL_OP独立链）及UT看护 Created-by: yuanbin_22 Commit-by: ywx1080114 Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动，包括改动的原因和所采取的方法。--> 作为 Infershape 整改的一部分，为 25 个 math 算子在 `op_graph/` 目录新增 InferDataType（输出数据类型推导）独立注册及 UT 看护，共 50 个文件（+1757 -0，纯新增），对齐 ops-nn 仓方案（参考 PR#9839 样板）：注册方案（IMPL_OP 独立链）： - 每个算子新增 `op_graph/<op>_graph_infer.cpp`，通过 `IMPL_OP(X).InferDataType(InferDataTypeX)` 独立注册链注册，与 `op_host` 的 `IMPL_OP_INFERSHAPE`（InferShape）链解耦； - 涉及 25 个算子：abs_grad、add_n、assign_add、assign_sub、cast、cos、erf、exp、expm1、floor、log1p、logical_not、masked_scale、nan_to_num、neg、reciprocal、rsqrt、rsqrt_grad、sign、sin、sqrt、sqrt_grad、square、tanh、trunc； - 原 35 个算子中的另外 10 个（atan、ceil、is_finite、is_inf、is_nan、is_neg_inf、is_pos_inf、log、right_shift、round）的 op_graph 迁移已单独合入 master，不在本 PR 范围；实现分类： - 24 个算子输出 dtype 与第一个输入一致（`SetOutputDataType(0, GetInputDataType(0))`）； - `Cast` 按属性推导并新增枚举范围校验：从 `gert::RuntimeAttrs` 读取 `dst_type` 属性，校验合法值域为 `[0, DT_MAX)` 且排除 `DT_UNDEFINED`（=28），非法值在推导处直接报错返回 `GRAPH_FAILED`（与 arch31 tiling 的 TransAclToGeDataType 白名单校验同款防御，防止非法值静默写入输出 desc）。 UT 看护（25 个测试文件，26 个用例）： - 每算子新增 `tests/ut/op_graph/test_<op>_graph_infer_pass.cpp`，直接调用导出的 `ops::InferDataTypeX` 函数，通过 `OpInferDataTypeContextBuilder` 构造上下文，参数化断言推导返回 `GRAPH_SUCCESS` 且输出 dtype 符合预期； - Cast 参数化覆盖 dst_type 集合（含 FP16/FLOAT/BF16/INT 系列等），并新增非法值专项用例：dst_type 为 -1/`DT_UNDEFINED`/`DT_MAX`/999 时断言推导返回 `GRAPH_FAILED`。 ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。--> <!-- 如果这个PR是为了解决特定的问题单，请在这里描述问题单单号。--> - 关联 Issue #3792 ## 测试 <!--描述进行了哪些测试来验证你的改动。包括但不限于二级冒烟、算子泛化等。--> - 25 个算子各新增 `InferDataTypeTest` 参数化 UT 用例（直接函数调用），全部通过； - Cast 额外新增非法 dst_type 专项用例（-1/DT_UNDEFINED/DT_MAX/999 断言 GRAPH_FAILED），覆盖枚举校验防御路径。 ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> - 无需文档更新。 ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!6392	11 days ago
op_host	feat: vector算子批量适配ascend960pr(DV122)平台 Co-authored-by: biabu111<hebaojing1@huawei.com> # message auto-generated for no-merge-commit merge: !6663 merge feat/dv122-ascend960pr-adaptation into master feat: vector算子批量适配ascend960pr(DV122)平台 Created-by: biabu111 Commit-by: biabu111 Merged-by: cann-robot Description: ## 描述 DV122（Ascend960PR / dav_9202 / Ascend960PR_969A1）vector 算子批量适配。参照 ascend960dt(DAV_9201) 既有适配模式，为 vector 算子清单内 66 个算子补齐 ascend960pr 支持，适配原则：9201 支持的能力 9202 均支持。注册级适配（66 算子）： - `CMakeLists.txt`：SUPPORT_COMPUTE_UNIT 追加 "ascend960pr"，SUPPORT_TILING_DIR 按位插入对应 arch35 - `op_host/_def.cpp`：`AICore().AddConfig("ascend960pr", <960dt 同款配置>)` - `op_host/config/ascend960pr/`：二进制配置（_binary.json / _simplified_key.ini，复制自 ascend960dt，与 960dt 内容一致） - `op_kernel/CMakeLists.txt`：COMPUTE_UNITS 追加 ascend960pr（arch92 kernel 入口同步生效） - `scripts/kernel/binary_config/ascendc_config.json`：compute_units 与 compile_options 补齐代码级补齐（9201 → 9201\|\|9202）： - tiling/aclnn/infershape/fusion_pass/kernel 中 `NpuArch::DAV_9201`、`__NPU_ARCH__ == 9201`、`"Ascend960DT"` 等 960 平台门禁补齐 `DAV_9202` / `9202` / `"Ascend960PR"` - 覆盖 concat/tanh_grad/slice/strided_slice/expand/tensor_equal/lin_space/range 等的 tiling/aclnn/graph_infer/fusion_pass 平台门禁（家族特性判定依据：gather_v2 适配提交中 isAscend96_ = DAV_9201\|\|DAV_9202、unsorted_segment_sum 注释"ascend960dt/ascend960pr(DAV_9201/DAV_9202) 平台判定，用于 960 专属特性(ScatterDataCopy 等)的运行时开关"） ## 关联的Issue https://gitcode.com/cann/ops-math/issues/4682 <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。例如：关联Issue #000--> ## 测试冒烟、ut、st已通过 <!--描述进行了哪些测试来验证你的改动。--> - CANN 9.3.0 工具链 `--soc=ascend960pr` 构建 19 个抽样算子（覆盖全部修改类型）：宿主侧（def/tiling/infershape/aclnn）100% 编译通过，0 错误 - 构建日志确认正确消费新增的 `config/ascend960pr/*_binary.json` - `--soc=ascend960dt` 同算子集构建行为一致，无回归 - 一致性审计：范围内 960dt-without-960pr 注册缺口 0；修改过的 CMakeLists SUPPORT_COMPUTE_UNIT/TILING_DIR 位置对齐 0 问题；ascendc_config.json 语法有效 - 注：opc 二进制生成阶段因当前工具链缺 Ascend960PR_969A1 平台数据无法本地执行，需在 CI（含 960PR 平台数据的工具链）完成二进制编译与上板验证 ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> 不涉及 ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： ## AI/Agent生成声明 <!-- [x] 表示选中 --> - [x] AI辅助编写 See merge request: cann/ops-math!6663	3 days ago
op_kernel	abs等算子A5实现 Co-authored-by: ly_cann_coder<liyao152@huawei.com> # message auto-generated for no-merge-commit merge: !495 merge develop into master abs等算子A5实现 Created-by: ly_cann_coder Commit-by: ly_cann_coder Merged-by: cann-robot Description: ## 描述新增以下算子的A5实现 \|序号\| 算子 \| 功能 \| \|--\|--\|--\| \|1\| abs \| 为输入张量的每一个元素取绝对值 \| \|2\| add \| 对输入完成相加操作 \| \|3\| addcdiv \| 张量运算函数，用于执行乘除加组合操作，将张量除法（带缩放）+ 张量加法合并为单个操作 \| \|4\| addcmul \| 张量运算函数，用于执行乘除加组合操作，将张量乘法（带缩放）+ 张量加法合并为单个操作 \| \|5\| axpy \| 源操作数中每个元素与标量求积后和目的操作数中的对应元素相加 \| \|6\| bincount \| 计算非负整数数组中每个数的频率 \| \|7\| ceil \| 返回输入tensor中每个元素向上取整的结果 \| \|8\| clip_by_value_v2 \| 将输入的所有元素限制在一定范围内 \| \|9\| concat \| 用于沿指定维度将多个输入 Tensor 进行拼接，输出包含所有输入数据按顺序拼接后的 Tensor \| \|10\| concat_v2 \| 用于沿指定维度将多个输入 Tensor 进行拼接，输出包含所有输入数据按顺序拼接后的 Tensor \| \|11\| equal \| 计算两个Tensor是否有相同的大小和元素，返回一个Bool类型 \| \|12\|exp\|返回一个新的张量，该张量的每个元素都是输入张量对应元素的指数\| \|13\| fill \| 对张量进行填充操作，支持非连续的Tensor操作 \| \|14\| floor \| 返回输入Tensor中每个元素向下取整，并将结果回填到输入Tensor中 \| \|15\| floor_div \| 完成除法计算，对余数向下取整 \| \|16\| floor_mod \| 用于计算两个数的模 \| \|17\| is_neg_inf \| 判断输入张量的元素是否为负无穷 \| \|18\| is_pos_inf \| 判断输入张量的元素是否为正无穷 \| \|19\| less \| 判断输入self中的每个元素是否小于输入other的值，返回一个Bool类型的Tensor \| \|20\| less_equal \| 判断输入self中的元素值是否小于等于other的值，并将self的每个元素的值与other值的比较结果写入out中 \| \|21\| logical_and \| 对两个输入张量的对应元素执行「与逻辑」判断，输出布尔型张量（True/False） \| \|22\| log \| 对输入张量x的元素，逐元素进行对数计算，并将结果保存到输出张量y中 \| \|23\| not_equal \| 逐元素比较两个输入张量是否不相等 \| \|24\| ones_like \| 返回形状和类型相同的张量，所有元素都设置为1 \| \|25\|sign\|对输入的tensor逐元素进行Sign符号函数的运算并输出结果tensor\| \|26\| sqrt \| 完成非负数平方根计算，负数情况返回nan \| \|27\| sqrt_grad \| 完成梯度平方根计算 \| \|28\| square \| 为输入张量的每一个元素计算平方值 \| \|29\| squared_difference \| 为第一个输入张量减去第二个输入张量，并计算其平方值 \| \|30\| tanh \| 激活函数，返回与输入tensor shape相同的tensor，对输入tensor进行elementwise的计算 \| \|31\| tril \| 将输入的self张量的最后二维（按shape从左向右数）沿对角线的右上部分置零 \| \|32\| triu \| 将输入的self张量的最后二维（按shape从左向右数）沿对角线的左下部分置零 \| \|33\| zeros_like \| 将张量填充为全零 \| ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。--> <!-- 如果这个PR是为了解决特定的问题单，请在这里描述问题单单号。--> ## 测试 <!--描述进行了哪些测试来验证你的改动。包括但不限于二级冒烟、算子泛化等。--> ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [x] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!495	10 months ago
op_kernel_aicpu	fix(aicpu): 补齐手写json缺失的opInfo字段 Co-authored-by: liu-wei<lovline.liuwei@huawei.com> # message auto-generated for no-merge-commit merge: !6519 merge fix/aicpu-opinfo-metadata into master fix(aicpu): 补齐手写json缺失的opInfo字段 Created-by: liu-wei Commit-by: liu-wei Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动，包括改动的原因、所采取的方法。--> 对齐 AICPU 整包生成侧与 CANN 9.2.0 安装版 `aicpu_kernel.json` 的 opInfo 元数据，补齐 6 个算子的字段缺失： - `conversion/identity/op_kernel_aicpu/identity.json` - 补齐 `formatAgnostic=True`、`opsFlag=OPS_FLAG_OPEN`（对齐安装版） - kernelSo 指向修正（libnn → libmath）已由 master db84400e2 率先合入，本 PR rebase 后不再重复改动 - `math/rsqrt/op_kernel_aicpu/rsqrt.json`、`math/sqrt/op_kernel_aicpu/sqrt.json`：补齐 `formatAgnostic=True`（对齐安装版） - `math/tile/op_kernel_aicpu/tile.json`：补齐 `opsFlag=OPS_FLAG_OPEN`（对齐安装版及同族 TileWithAxis 注册写法） - `math/pow/op_kernel_aicpu/pow.json`：补齐 `opsFlag=OPS_FLAG_CLOSE`（对齐安装版） - `math/greater_equal/op_kernel_aicpu/greater_equal_aicpu_def.cpp`：补齐 `TRUE_FORMAT_AGNOSTIC` 覆盖（安装版比较家族 GreaterEqual/Less/NotEqual 均为 True、Equal 为 False，本仓 Less/NotEqual 已覆盖，唯独 GreaterEqual 漏了 formatAgnostic）背景：三仓 AICPU 整包 json 与安装版逐算子逐字段比对（[cann/ops-cv#1276](https://gitcode.com/cann/ops-cv/issues/1276)）定位到上述差异。76 个 AICPU 算子中 5 个为手写 json（identity/rsqrt/sqrt/tile/pow），不走 `ApplyMathAicpuDefaultCfg`+`ExtendCfgInfo` 注册机制，迁移/手写时字段易漏，本 PR 逐一补齐；后续建议将手写 json 逐步迁移为 def 注册，从根本上规避。 ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。--> <!-- 如果这个PR是为了解决特定的问题单，请在这里描述问题单单号。--> [cann/ops-cv#1276](https://gitcode.com/cann/ops-cv/issues/1276)（AICPU整包json与安装版json 算子定义差异梳理） ## 测试 <!--描述进行了哪些测试来验证你的改动。包括但不限于二级冒烟、算子泛化等。--> - 5 个手写 json 语法校验通过 - 76 个 AICPU 算子源码推导值与本机 CANN 9.2.0 安装版全量比对：修复后 6 个目标算子的 formatAgnostic/opsFlag 与安装版完全对齐，其余 3 项差异（Coordinates1DTo2D/OnesLike/TileWithAxis，整包侧演进项）保持不变 - trans_soc / find_value_by_key 等既有机制无行为影响（GreaterEqual 仅新增一行 ExtendCfgInfo 覆盖） - pre-commit 全钩子通过（json 校验 / clang-format / codespell / OAT 合规） - 重建验证：重新 configure（重编 71 个 opdef 生成 aicpu_kernel.ini）并构建 gen_aicpu_json_from_opdef，再生成 aicpu_math.json——76 算子齐全，6 个目标算子的 kernelSo/formatAgnostic/opsFlag 全部落盘正确（Identity kernelSo 由 libnn 修正为 libmath，三处 True/OPEN 补齐，GreaterEqual 经 def→ini→json 链路生效），全量比对无回归 - 符号核验：构建 libmath_aicpu_kernels.so（aarch64），RunCpuKernel 动态符号导出、"Identity" 注册名字符串在 so 内，非本仓 AICPU 算子名（Conv2D/ROIAlign 等）反证 0 命中 - 未执行 NPU 上板运行与完整打包封测（无硬件环境；CUST 注册加载行为需端到端图编译验证） - rebase 至 master f597aafa 后复核：剩余字段补齐改动与原 PR 逐行一致，master db84400e2 已含 kernelSo 修正（本 PR 该项变为无差异），pre-commit 全钩子通过 ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新README.md文件。--> 无 ## 类型标签 <!-- [x] 表示选中 --> - [x] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!6519	17 days ago
tests	feat: Infershape整改，25个math算子InferDataType注册迁移至op_graph目录（IMPL_OP独立链）及UT看护 Co-authored-by: ywx1080114<yuanbin22@huawei.com> # message auto-generated for no-merge-commit merge: !6392 merge infershape-dtype into master feat: Infershape整改，25个math算子InferDataType注册迁移至op_graph目录（IMPL_OP独立链）及UT看护 Created-by: yuanbin_22 Commit-by: ywx1080114 Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动，包括改动的原因和所采取的方法。--> 作为 Infershape 整改的一部分，为 25 个 math 算子在 `op_graph/` 目录新增 InferDataType（输出数据类型推导）独立注册及 UT 看护，共 50 个文件（+1757 -0，纯新增），对齐 ops-nn 仓方案（参考 PR#9839 样板）：注册方案（IMPL_OP 独立链）： - 每个算子新增 `op_graph/<op>_graph_infer.cpp`，通过 `IMPL_OP(X).InferDataType(InferDataTypeX)` 独立注册链注册，与 `op_host` 的 `IMPL_OP_INFERSHAPE`（InferShape）链解耦； - 涉及 25 个算子：abs_grad、add_n、assign_add、assign_sub、cast、cos、erf、exp、expm1、floor、log1p、logical_not、masked_scale、nan_to_num、neg、reciprocal、rsqrt、rsqrt_grad、sign、sin、sqrt、sqrt_grad、square、tanh、trunc； - 原 35 个算子中的另外 10 个（atan、ceil、is_finite、is_inf、is_nan、is_neg_inf、is_pos_inf、log、right_shift、round）的 op_graph 迁移已单独合入 master，不在本 PR 范围；实现分类： - 24 个算子输出 dtype 与第一个输入一致（`SetOutputDataType(0, GetInputDataType(0))`）； - `Cast` 按属性推导并新增枚举范围校验：从 `gert::RuntimeAttrs` 读取 `dst_type` 属性，校验合法值域为 `[0, DT_MAX)` 且排除 `DT_UNDEFINED`（=28），非法值在推导处直接报错返回 `GRAPH_FAILED`（与 arch31 tiling 的 TransAclToGeDataType 白名单校验同款防御，防止非法值静默写入输出 desc）。 UT 看护（25 个测试文件，26 个用例）： - 每算子新增 `tests/ut/op_graph/test_<op>_graph_infer_pass.cpp`，直接调用导出的 `ops::InferDataTypeX` 函数，通过 `OpInferDataTypeContextBuilder` 构造上下文，参数化断言推导返回 `GRAPH_SUCCESS` 且输出 dtype 符合预期； - Cast 参数化覆盖 dst_type 集合（含 FP16/FLOAT/BF16/INT 系列等），并新增非法值专项用例：dst_type 为 -1/`DT_UNDEFINED`/`DT_MAX`/999 时断言推导返回 `GRAPH_FAILED`。 ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。--> <!-- 如果这个PR是为了解决特定的问题单，请在这里描述问题单单号。--> - 关联 Issue #3792 ## 测试 <!--描述进行了哪些测试来验证你的改动。包括但不限于二级冒烟、算子泛化等。--> - 25 个算子各新增 `InferDataTypeTest` 参数化 UT 用例（直接函数调用），全部通过； - Cast 额外新增非法 dst_type 专项用例（-1/DT_UNDEFINED/DT_MAX/999 断言 GRAPH_FAILED），覆盖枚举校验防御路径。 ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> - 无需文档更新。 ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!6392	11 days ago
CMakeLists.txt	feat: vector算子批量适配ascend960pr(DV122)平台 Co-authored-by: biabu111<hebaojing1@huawei.com> # message auto-generated for no-merge-commit merge: !6663 merge feat/dv122-ascend960pr-adaptation into master feat: vector算子批量适配ascend960pr(DV122)平台 Created-by: biabu111 Commit-by: biabu111 Merged-by: cann-robot Description: ## 描述 DV122（Ascend960PR / dav_9202 / Ascend960PR_969A1）vector 算子批量适配。参照 ascend960dt(DAV_9201) 既有适配模式，为 vector 算子清单内 66 个算子补齐 ascend960pr 支持，适配原则：9201 支持的能力 9202 均支持。注册级适配（66 算子）： - `CMakeLists.txt`：SUPPORT_COMPUTE_UNIT 追加 "ascend960pr"，SUPPORT_TILING_DIR 按位插入对应 arch35 - `op_host/_def.cpp`：`AICore().AddConfig("ascend960pr", <960dt 同款配置>)` - `op_host/config/ascend960pr/`：二进制配置（_binary.json / _simplified_key.ini，复制自 ascend960dt，与 960dt 内容一致） - `op_kernel/CMakeLists.txt`：COMPUTE_UNITS 追加 ascend960pr（arch92 kernel 入口同步生效） - `scripts/kernel/binary_config/ascendc_config.json`：compute_units 与 compile_options 补齐代码级补齐（9201 → 9201\|\|9202）： - tiling/aclnn/infershape/fusion_pass/kernel 中 `NpuArch::DAV_9201`、`__NPU_ARCH__ == 9201`、`"Ascend960DT"` 等 960 平台门禁补齐 `DAV_9202` / `9202` / `"Ascend960PR"` - 覆盖 concat/tanh_grad/slice/strided_slice/expand/tensor_equal/lin_space/range 等的 tiling/aclnn/graph_infer/fusion_pass 平台门禁（家族特性判定依据：gather_v2 适配提交中 isAscend96_ = DAV_9201\|\|DAV_9202、unsorted_segment_sum 注释"ascend960dt/ascend960pr(DAV_9201/DAV_9202) 平台判定，用于 960 专属特性(ScatterDataCopy 等)的运行时开关"） ## 关联的Issue https://gitcode.com/cann/ops-math/issues/4682 <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。例如：关联Issue #000--> ## 测试冒烟、ut、st已通过 <!--描述进行了哪些测试来验证你的改动。--> - CANN 9.3.0 工具链 `--soc=ascend960pr` 构建 19 个抽样算子（覆盖全部修改类型）：宿主侧（def/tiling/infershape/aclnn）100% 编译通过，0 错误 - 构建日志确认正确消费新增的 `config/ascend960pr/*_binary.json` - `--soc=ascend960dt` 同算子集构建行为一致，无回归 - 一致性审计：范围内 960dt-without-960pr 注册缺口 0；修改过的 CMakeLists SUPPORT_COMPUTE_UNIT/TILING_DIR 位置对齐 0 问题；ascendc_config.json 语法有效 - 注：opc 二进制生成阶段因当前工具链缺 Ascend960PR_969A1 平台数据无法本地执行，需在 CI（含 960PR 平台数据的工具链）完成二进制编译与上板验证 ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> 不涉及 ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： ## AI/Agent生成声明 <!-- [x] 表示选中 --> - [x] AI辅助编写 See merge request: cann/ops-math!6663	3 days ago
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 month ago
README_en.md	docs: add English translations for markdown files Co-authored-by: jenny_xiang<jenny.xiang@huawei.com> # message auto-generated for no-merge-commit merge: !7113 merge docs/add-english-translations into master docs: add English translations for markdown files Created-by: jenny_xiang Commit-by: jenny_xiang Merged-by: cann-robot Description: ## 描述新增仓库Markdown文档英文译稿（第二批，共409个文件）：第一批（已合入）之外的其余全部译稿，覆盖conversion、examples、ge、math、random各类算子README译稿，以及CHANGELOG_en.md、.gitcode PR模板英文版等。本批内全部仓内链接均指向master已有文件（含第一批已合入的21+2篇译稿），合入后本仓432篇英文文档全部就位，链接完整生效。本批同时包含4篇基于最新源文档的增量更新（math/ada_cast、math/prod_force_se_a、math/sinkhorn、math/xlogy，对应上游近期文档变更）。 ## 关联的Issue 无 ## 测试文档类变更，无需算子测试；已通过链接有效性校验（本批全部相对链接仅指向master已有文件，无指向本PR外未提交文件的链接），并完成逐篇翻译规范自查与语义校对（含支持状态表、限定词、术语、数值符号核对）。 ## 文档更新新增409个英文译稿文件（*_en.md / .en-US.md），不修改任何现有文件。 ## 类型标签 - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [x] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!7113	9 days ago

Sqrt

English | 简体中文

Product Support Status

Product	Supported
Ascend 950PR&950DT products	√
Atlas A3 products	√
Atlas A2 products	√
Atlas 200I/500 A2 inference products	×
Atlas inference products	×
Atlas training products	√

Function Description

Operator function: computes the square root of non-negative numbers, and returns nan for negative numbers.
Computation formula:
$out=sqrt(self)={self,self≥0,nan,self<0out=sqrt(self)=\begin{cases} \sqrt {self}, & self\ge 0 , \\ nan, & self\lt 0 \end{cases}$

Parameter Description

Parameter Name	Input/Output/Attribute	Description	Data Type	Data Format
self	Input	Input parameter for the sqrt computation, which is self in the formula.	COMPLEX64, COMPLEX128, FLOAT, FLOAT16, BFLOAT16, DOUBLE, INT32, INT64, INT16, INT8, UINT8, BOOL	ND
out	Output	Output parameter for the sqrt computation, which is out in the formula.	FLOAT, FLOAT16, DOUBLE, COMPLEX64, COMPLEX128, BFLOAT16	ND

Atlas training products and Atlas inference products: BFLOAT16 is not supported.

Constraint Description

None

Invocation Description

Invocation Method	Invocation Sample	Description
aclnn invocation	test_aclnn_sqrt	Invokes the Sqrt operator through the aclnnSqrt API.
Graph-mode invocation	test_geir_sqrt	Invokes the Sqrt operator by building a graph through the operator IR.