cann-robotDynamicQuant等22个算子支持Ascend960dt

Files	Last commit	Last update
docs	删除源码链接 Co-authored-by: fwx1169505<fengying27@h-partners.com> # message auto-generated for no-merge-commit merge: !12106 merge master into master 删除源码链接 Created-by: fengying555682 Commit-by: fwx1169505 Merged-by: cann-robot Description: ## 描述删除源码链接 ## 关联的Issue [#6769](https://gitcode.com/cann/ops-nn/issues/6769) ## 测试仅涉及文档修改，已经测试md文档语法 ## 文档更新 activation和quant目录下的md文档 ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ x] 文档更新 - [ ] 其他，请描述： ## AI/Agent生成声明 <!-- [x] 表示选中 --> - [ ] AI辅助编写 See merge request: cann/ops-nn!12106	24 days ago
examples	fix: 修复7个算子文档示例代码静态检查问题 Co-authored-by: shilulu<shilulu8@huawei.com> # message auto-generated for no-merge-commit merge: !10472 merge clean_code into master fix: 修复7个算子文档示例代码静态检查问题 Created-by: shilulu Commit-by: shilulu Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动，包括改动的原因和所采取的方法。--> fix: 修复7个算子文档示例代码静态检查问题 ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。例如：关联Issue #000--> https://gitcode.com/cann/ops-nn/issues/5901 ## 测试 <!--描述进行了哪些测试来验证你的改动。--> ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> 更新了： aclnnHardsigmoid&aclnnInplaceHardsigmoid.md aclnnLogSoftmax.md aclnnSeluBackward.md aclnnSigmoid&aclnnInplaceSigmoid.md（activation） aclnnSigmoid&aclnnInplaceSigmoid.md（experimental） aclnnSwiGluGrad.md aclnnGroupNormSilu.md aclnnGroupNormSiluV2.md ## 类型标签 <!-- [x] 表示选中 --> - [x] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [x] 文档更新 - [ ] 其他，请描述： ## AI/Agent生成声明 <!-- [x] 表示选中 --> - [ ] AI辅助编写 See merge request: cann/ops-nn!10472	1 month ago
framework	TF plugin开源 Co-authored-by: sunchun<sunchun4@h-partners.com> # message auto-generated for no-merge-commit merge: !7516 merge nn into master TF plugin开源 Created-by: sunchun Commit-by: sunchun Merged-by: cann-robot Description: ## 描述 TF plugin开源修改 ## 关联的Issue https://gitcode.com/cann/ops-nn/issues/4196 ## 测试 <!--描述进行了哪些测试来验证你的改动。--> ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： ## AI/Agent生成声明 <!-- [x] 表示选中 --> - [ ] AI辅助编写 See merge request: cann/ops-nn!7516	3 months ago
op_api	feat: 增加 ascend350 平台支持（25 个激活/量化算子）并重构 SocVersion 判断为 NpuArch Co-authored-by: wenqi_wu<wenqiwu0628@163.com> # message auto-generated for no-merge-commit merge: !9834 merge support-ascend350-activation-ops into master feat: 增加 ascend350 平台支持（25 个激活/量化算子）并重构 SocVersion 判断为 NpuArch Created-by: wenqi_wu Commit-by: wenqi_wu Merged-by: cann-robot Description: ## 描述为以下 25 个算子增加 ascend350（arch35）平台支持，并将 7 个 aclnn 接口的 SocVersion 判断重构为 NpuArch 判断。算子列表（ascend350 支持，25 个）： - 量化类：AscendQuant / AscendAntiQuant / Quantize（quant） - Elu / EluGrad / EluGradV2（activation） - FastGelu / FastGeluGrad - Gelu / GeluGrad / GeluV2 / GeluGradV2 - ReluGrad / ReluGradV2 / ReluV2 - LeakyReluGrad - SigmoidGrad / SiluGrad - Swish / SwishGrad - HardtanhGrad - Mish / MishGrad - LogSigmoid / LogSigmoidGrad 注：LeakyRelu 的 ascend350 支持已由上游先行合入，本 PR 不含其重复改动。改动内容： 1. 算子 CMakeLists.txt：SUPPORT_COMPUTE_UNIT 追加 ascend350，SUPPORT_TILING_DIR 追加对应 arch35； 2. op_host/_def.cpp：为每个算子追加 this->AICore().AddConfig("ascend350", ...)； 3. 新增 op_host/config/ascend350/_binary.json 二进制配置文件； 4. scripts/kernel/binary_config/ascendc_config.json：为上述算子增加 ascend350 的 compute_units 与 compile_options（AscendAntiQuant 上游本无条目，走默认编译选项）； 5. op_api 中 SocVersion → NpuArch 重构（7 处）： - aclnn_rrelu_with_noise.cpp：GenerateUniformRandom 中平台判断改为 GetCurNpuArch() == NpuArch::DAV_2201（语义等价，910B/910_93 均对应 DAV_2201）； - aclnn_sigmoid_backward.cpp / aclnn_swish_backward.cpp / aclnn_gelu_v2.cpp：CheckSocVersionIsSupportBf16 改为 curArch == NpuArch::DAV_2201 \|\| IsRegbase(curArch)； - aclnn_sigmoid.cpp：CheckSocVersionIsSupportBf16 与 GetSelfRefDtypeList 改为 curArch == NpuArch::DAV_2201 \|\| IsRegbase(curArch)； - aclnn_binary_cross_entropy_with_logits_target_backward.cpp：GetDtypeSupportList 改为 NpuArch 判断； - aclnn_logsigmoid_backward.cpp：GetInputDtypeSupportList / GetOutputDtypeSupportList 改为 NpuArch 判断。 ## 关联的Issue https://gitcode.com/cann/ops-nn/issues/5538 ## 测试 <!--TODO--> ## 文档更新无需更新文档 ## 类型标签 - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其它（请说明） ## AI/Agent辅助编译 - [x] AI辅助编写 See merge request: cann/ops-nn!9834	1 month ago
op_graph	format cpp Co-authored-by: yang-di52<yangdi52@huawei.com> # message auto-generated for no-merge-commit merge: !6784 merge issue_fix into master format cpp Created-by: yang-di52 Commit-by: yang-di52 Merged-by: cann-robot Description: ## 描述批量刷新cpp代码格式 ## 关联的Issue [#3791](https://gitcode.com/cann/ops-nn/issues/3791) ## 测试 <!--描述进行了哪些测试来验证你的改动。--> ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [x] 其他，请描述：代码格式化 ## AI/Agent生成声明 <!-- [x] 表示选中 --> - [ ] AI辅助编写 See merge request: cann/ops-nn!6784	4 months ago
op_host	DynamicQuant等22个算子支持Ascend960dt Co-authored-by: TH_HW<taohai2@huawei.com> # message auto-generated for no-merge-commit merge: !11335 merge backBlue into master DynamicQuant等22个算子支持Ascend960dt Created-by: TH_HW Commit-by: TH_HW Merged-by: cann-robot Description: ## 描述（+17812 / -2151）为一批 ops-nn 算子补齐 Ascend 960DT（NPU Arch 9201）平台支持，并重构 GeGluV2 arch35 内核、为 Erfinv 增加 inplace 通路。主要改动如下： 1. Ascend960DT 平台适配（22 个算子） - `activation`：Relu、ReluV2、ReluGrad、ReluGradV2、Sigmoid、SigmoidGrad、PRelu、PReluGradUpdate、SiluGrad、Softplus、Swish、SwishGrad、SwiGluGrad、GeGluV2。 - `index`：MapIndex、SparseSlice、QuantUpdateScatter。 - `quant`：Quantize、DynamicQuant、DynamicQuantV2、AntiMxQuant、GroupedDynamicMxQuant。 - 具体做法：`CMakeLists.txt` 的 `SUPPORT_COMPUTE_UNIT` / `SUPPORT_TILING_DIR` 增加 `ascend960dt`；`_def.cpp` 增加 `AddConfig("ascend960dt", ...)`；aclnn/op_api 与 infershape 增加 `NpuArch::DAV_9201` / `Ascend960DT` 分支；kernel 增加 `__NPU_ARCH__ == 9201` 分支；新增 `op_host/config/ascend960dt/` 下的算子二进制配置（共 22 个，如 `relu_binary.json`、`p_relu.json` 等）；`scripts/kernel/binary_config/ascendc_config.json` 为相关算子登记 ascend960dt 计算单元与编译选项。 2. GeGluV2 arch35 内核重构* - 将 arch35 模板按 `erf` / `tanh` 两类拆分，并补齐 `align`、`align_last_axis_big`、`vreduce` × `fp16/bf16/fp32` 变体，统一命名 `_arch35.h`。 - tiling 新增 RegBase per-template per-element UB 预算模型（`GetRegbasePerElem`），并支持 ascend960dt。 - `aclnnGeGlu` 参数校验重构（`CheckShape` 支持 0D 输入与 dim 范围校验），新增 ST 用例（`all_aclnnGeGlu(.V3).json`）并调整 UT。 3. Erfinv 支持 inplace：新增 inplace dtype 支持列表与校验（`GetInplaceDtypeSupportList` / `CheckInplaceDtypeValid`），并补充 op_api UT。 4. 内核与接口修复* - `SparseSlice`：9201 路径改用 `SetValue` 写入 sliceStart/sliceEnd（替代 `DataCopy`），`Reg::Reduce` 使用完整命名空间限定。 - `QuantUpdateScatter`：9201 增加 `FLOAT_OVERFLOW_MODE_CTRL` 保护。 - `ReluV2` infershape：Ascend960DT 不支持 NC1HWC0，UINT1 支持列表加入 Ascend960DT。 - `DynamicQuant` / Mx 量化：9201 上 fp8 scale 改为 bf16 乘法后再 cast 到 e8m0。 5. 代码规范：统一 License 头注释格式、补齐文件末尾换行、调整 quant 系列 UT 的 CMakeLists；新增 `MapIndex`、`SparseSlice` 的 GE IR 图模式示例。 ## 关联的Issue 无。 ## 测试 - 新增 GeGluV2 ST 用例与 Erfinv inplace UT，补充/调整 GeGluV2 相关 UT。 - 新增 MapIndex / SparseSlice 的 GE IR 示例（`examples/test_geir_*.cpp`）。 - 通过 ascend960dt 编译与静态检查。 ## 文档更新无。 ## 类型标签 <!-- [x] 表示选中 --> - [x] 新特性 - [ ] Bug修复 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： ## AI/Agent生成声明 <!-- [x] 表示选中 --> - [x] AI辅助编写 See merge request: cann/ops-nn!11335	22 days ago
op_kernel	MicroAPI namespace to Reg for arch35 kernels Co-authored-by: gcw_DS4cmz2b<1312925094@qq.com> # message auto-generated for no-merge-commit merge: !9267 merge rename into master MicroAPI namespace to Reg for arch35 kernels Created-by: gcw_DS4cmz2b Commit-by: gcw_DS4cmz2b Merged-by: cann-robot Description: ## 描述修改了index,activation,matmul,conv,foreach,optim,experimental,vfusion,hash这几个文件夹内的命名空间，将MicroAPI-->Reg ## 关联的Issue [多个文件夹内的算子命名空间使用的MicroAPI，应改为Reg](https://gitcode.com/cann/ops-nn/issues/5218) ## 测试验证方法严格按四步流程，逐算子验证： 1. Step 1：编译原始代码，保存 baseline binary .o 文件 MD5 2. Step 2：执行 MicroAPI → Reg 替换 3. Step 3：编译替换后代码，保存 after binary .o 文件 MD5 4. Step 4：对比 A_before.md5 与 B_after.md5 平台：ascend950，编译参数：--soc=ascend950 --ops=<op> -j16 验证替换前后 binary 产物 MD5 完全一致。冒烟，代码审查均通过 ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： ## AI/Agent生成声明 <!-- [x] 表示选中 --> - [ ] AI辅助编写 See merge request: cann/ops-nn!9267	2 months ago
tests	sigmoid的ws改成0 Co-authored-by: ligen<ligen75@h-partners.com> # message auto-generated for no-merge-commit merge: !9804 merge master into master sigmoid的ws改成0 Created-by: ligen75 Commit-by: ligen Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动，包括改动的原因和所采取的方法。--> sigmoid的系统workspace改成0 ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。例如：关联Issue #000--> https://gitcode.com/cann/ops-nn/issues/5482 ## 测试 <!--描述进行了哪些测试来验证你的改动。--> 自验通过 ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： ## AI/Agent生成声明 <!-- [x] 表示选中 --> - [ ] AI辅助编写 See merge request: cann/ops-nn!9804	2 months ago
CMakeLists.txt	DynamicQuant等22个算子支持Ascend960dt Co-authored-by: TH_HW<taohai2@huawei.com> # message auto-generated for no-merge-commit merge: !11335 merge backBlue into master DynamicQuant等22个算子支持Ascend960dt Created-by: TH_HW Commit-by: TH_HW Merged-by: cann-robot Description: ## 描述（+17812 / -2151）为一批 ops-nn 算子补齐 Ascend 960DT（NPU Arch 9201）平台支持，并重构 GeGluV2 arch35 内核、为 Erfinv 增加 inplace 通路。主要改动如下： 1. Ascend960DT 平台适配（22 个算子） - `activation`：Relu、ReluV2、ReluGrad、ReluGradV2、Sigmoid、SigmoidGrad、PRelu、PReluGradUpdate、SiluGrad、Softplus、Swish、SwishGrad、SwiGluGrad、GeGluV2。 - `index`：MapIndex、SparseSlice、QuantUpdateScatter。 - `quant`：Quantize、DynamicQuant、DynamicQuantV2、AntiMxQuant、GroupedDynamicMxQuant。 - 具体做法：`CMakeLists.txt` 的 `SUPPORT_COMPUTE_UNIT` / `SUPPORT_TILING_DIR` 增加 `ascend960dt`；`_def.cpp` 增加 `AddConfig("ascend960dt", ...)`；aclnn/op_api 与 infershape 增加 `NpuArch::DAV_9201` / `Ascend960DT` 分支；kernel 增加 `__NPU_ARCH__ == 9201` 分支；新增 `op_host/config/ascend960dt/` 下的算子二进制配置（共 22 个，如 `relu_binary.json`、`p_relu.json` 等）；`scripts/kernel/binary_config/ascendc_config.json` 为相关算子登记 ascend960dt 计算单元与编译选项。 2. GeGluV2 arch35 内核重构* - 将 arch35 模板按 `erf` / `tanh` 两类拆分，并补齐 `align`、`align_last_axis_big`、`vreduce` × `fp16/bf16/fp32` 变体，统一命名 `_arch35.h`。 - tiling 新增 RegBase per-template per-element UB 预算模型（`GetRegbasePerElem`），并支持 ascend960dt。 - `aclnnGeGlu` 参数校验重构（`CheckShape` 支持 0D 输入与 dim 范围校验），新增 ST 用例（`all_aclnnGeGlu(.V3).json`）并调整 UT。 3. Erfinv 支持 inplace：新增 inplace dtype 支持列表与校验（`GetInplaceDtypeSupportList` / `CheckInplaceDtypeValid`），并补充 op_api UT。 4. 内核与接口修复* - `SparseSlice`：9201 路径改用 `SetValue` 写入 sliceStart/sliceEnd（替代 `DataCopy`），`Reg::Reduce` 使用完整命名空间限定。 - `QuantUpdateScatter`：9201 增加 `FLOAT_OVERFLOW_MODE_CTRL` 保护。 - `ReluV2` infershape：Ascend960DT 不支持 NC1HWC0，UINT1 支持列表加入 Ascend960DT。 - `DynamicQuant` / Mx 量化：9201 上 fp8 scale 改为 bf16 乘法后再 cast 到 e8m0。 5. 代码规范：统一 License 头注释格式、补齐文件末尾换行、调整 quant 系列 UT 的 CMakeLists；新增 `MapIndex`、`SparseSlice` 的 GE IR 图模式示例。 ## 关联的Issue 无。 ## 测试 - 新增 GeGluV2 ST 用例与 Erfinv inplace UT，补充/调整 GeGluV2 相关 UT。 - 新增 MapIndex / SparseSlice 的 GE IR 示例（`examples/test_geir_*.cpp`）。 - 通过 ascend960dt 编译与静态检查。 ## 文档更新无。 ## 类型标签 <!-- [x] 表示选中 --> - [x] 新特性 - [ ] Bug修复 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： ## AI/Agent生成声明 <!-- [x] 表示选中 --> - [x] AI辅助编写 See merge request: cann/ops-nn!11335	22 days ago
README.md	修改产品名 Co-authored-by: fwx1169505<fengying27@h-partners.com> # message auto-generated for no-merge-commit merge: !11017 merge master into master 修改产品名 Created-by: fengying555682 Commit-by: fwx1169505 Merged-by: cann-robot Description: ## 描述修改quant和activation文件夹中的md文档涉及的产品名称 ## 关联的Issue [#6056](https://gitcode.com/cann/ops-nn/issues/6056) ## 测试已经完成文档语法和链接一致性检测 ## 文档更新 quant和activation文件夹中的md文档 ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [x ] 文档更新 - [ ] 其他，请描述： ## AI/Agent生成声明 <!-- [x] 表示选中 --> - [ ] AI辅助编写 See merge request: cann/ops-nn!11017	1 month ago

Sigmoid

产品支持情况

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

功能说明

算子功能：对输入Tensor完成Sigmoid运算。
计算公式：

{\frac{1} {1+{e}^{-input}}}

参数说明

参数名	输入/输出	描述	数据类型	数据格式
x	输入	待进行Sigmoid计算的入参，公式中的input。	FLOAT、FLOAT16、BFLOAT16	ND
y	输出	计算的出参。	FLOAT、FLOAT16、BFLOAT16	ND

约束说明

无

调用说明

调用方式	调用样例	说明
aclnn调用	test_aclnn_sigmoid	通过aclnnSigmoid&aclnnInplaceSigmoid接口方式调用Sigmoid算子。
aclnn调用	test_aclnn_inplace_sigmoid	通过aclnnSigmoid&aclnnInplaceSigmoid接口方式调用Sigmoid算子。
图模式	-	通过算子IR构图方式调用Sigmoid算子。