Star0

Star0

cann-robot修复并编译运行aclnn的示例

048a2d1c创建于 8月11日历史提交

文件	最后提交记录	最后更新时间
docs	打标签 Co-authored-by: y60124828<yanglou1@h-partners.com> # message auto-generated for no-merge-commit merge: !8369 merge master into master 打标签 Created-by: yanglu-1 Commit-by: y60124828 Merged-by: cann-robot Description: ## 描述 nn仓打标签 ## 关联的Issue 关联Issue [#4600](https://gitcode.com/cann/ops-nn/issues/4600) ## 测试 <!--描述进行了哪些测试来验证你的改动。--> ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [x] 文档更新 - [ ] 其他，请描述： ## AI/Agent生成声明 <!-- [x] 表示选中 --> - [ ] AI辅助编写 See merge request: cann/ops-nn!8369	1 个月前
examples	修复并编译运行aclnn的示例 Co-authored-by: xiaodong666<yuanxiaodong6@huawei.com> # message auto-generated for no-merge-commit merge: !8548 merge aclnnInplacePut_fix into master 修复并编译运行aclnn的示例 Created-by: xiaodong666 Commit-by: xiaodong666 Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动，包括改动的原因和所采取的方法。--> 修复并编译运行测test_aclnn_inplace_put.cpp、test_aclnn_scatter_nd_update.cpp、test_aclnn_scatter_nd.cpp、test_aclnn_index_copy.cpp、的示例 ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。例如：关联Issue #000--> 关联Issue [#4711](https://gitcode.com/cann/ops-nn/issues/4711) ## 测试 <!--描述进行了哪些测试来验证你的改动。--> 编译运行示例成功 ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> NA ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [x] 文档更新 - [ ] 其他，请描述： ## AI/Agent生成声明 <!-- [x] 表示选中 --> - [ ] AI辅助编写 See merge request: cann/ops-nn!8548	1 个月前
framework	TF plugin修改1 Co-authored-by: sunchun<sunchun4@h-partners.com> # message auto-generated for no-merge-commit merge: !7547 merge nn1 into master TF plugin修改1 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!7547	1 个月前
op_api	refactor: 统一 arch32 → arch22 命名（index/pooling/conv） Co-authored-by: chen-xing-yu<chenxingyu18@huawei.com> # message auto-generated for no-merge-commit merge: !8210 merge refactor/arch32_to_arch22 into master refactor: 统一 arch32 → arch22 命名（index/pooling/conv） Created-by: chenxingyu18 Commit-by: chen-xing-yu Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动，包括改动的原因和所采取的方法。--> 将 index/、pooling/、conv/ 下所有 arch32 目录与引用统一改为 arch22，与全仓主流命名约定对齐（arch22 用于 ascend910b/910_93，arch35 用于 ascend950）。 - 目录重命名 arch32 → arch22（op_host/op_tiling、op_kernel、tests/ut 等） - 文件内容：include 路径、CMakeLists 的 SUPPORT_TILING_DIR、注释中的 arch32 → arch22 - arch35 目录与引用未受影响 - 顺带修复两个被禁用（main() 提前 return 0）的 eager example：index_add、scatter_nd_update ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。例如：关联Issue #000--> #4493 ## 测试 <!--描述进行了哪些测试来验证你的改动。--> 基于 ascend910b（CANN 9.1.0 / 910B3）完成统一打包构建 + eager 验证，涉及算子 index_fill、scatter_nd_update、masked_scatter、adaptive_max_pool3d(_grad)、max_pool3d_grad_with_argmax、avg_pool3_d、inplace_index_add_with_sorted、conv3d_backprop_filter_v2、conv3d_backprop_input_v2、conv3d_transpose_v2 均验证通过；并用改名测试（移除 vendor kernel 后 example 失败）确认跑的是改名后的实现而非 stock CANN。 ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [x] 其他，请描述：命名统一（arch32 → arch22） ## AI/Agent生成声明 <!-- [x] 表示选中 --> - [x] AI辅助编写 See merge request: cann/ops-nn!8210	1 个月前
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	2 个月前
op_host	refactor: 统一 arch32 → arch22 命名（index/pooling/conv） Co-authored-by: chen-xing-yu<chenxingyu18@huawei.com> # message auto-generated for no-merge-commit merge: !8210 merge refactor/arch32_to_arch22 into master refactor: 统一 arch32 → arch22 命名（index/pooling/conv） Created-by: chenxingyu18 Commit-by: chen-xing-yu Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动，包括改动的原因和所采取的方法。--> 将 index/、pooling/、conv/ 下所有 arch32 目录与引用统一改为 arch22，与全仓主流命名约定对齐（arch22 用于 ascend910b/910_93，arch35 用于 ascend950）。 - 目录重命名 arch32 → arch22（op_host/op_tiling、op_kernel、tests/ut 等） - 文件内容：include 路径、CMakeLists 的 SUPPORT_TILING_DIR、注释中的 arch32 → arch22 - arch35 目录与引用未受影响 - 顺带修复两个被禁用（main() 提前 return 0）的 eager example：index_add、scatter_nd_update ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。例如：关联Issue #000--> #4493 ## 测试 <!--描述进行了哪些测试来验证你的改动。--> 基于 ascend910b（CANN 9.1.0 / 910B3）完成统一打包构建 + eager 验证，涉及算子 index_fill、scatter_nd_update、masked_scatter、adaptive_max_pool3d(_grad)、max_pool3d_grad_with_argmax、avg_pool3_d、inplace_index_add_with_sorted、conv3d_backprop_filter_v2、conv3d_backprop_input_v2、conv3d_transpose_v2 均验证通过；并用改名测试（移除 vendor kernel 后 example 失败）确认跑的是改名后的实现而非 stock CANN。 ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [x] 其他，请描述：命名统一（arch32 → arch22） ## AI/Agent生成声明 <!-- [x] 表示选中 --> - [x] AI辅助编写 See merge request: cann/ops-nn!8210	1 个月前
op_kernel	refactor: 统一 arch32 → arch22 命名（index/pooling/conv） Co-authored-by: chen-xing-yu<chenxingyu18@huawei.com> # message auto-generated for no-merge-commit merge: !8210 merge refactor/arch32_to_arch22 into master refactor: 统一 arch32 → arch22 命名（index/pooling/conv） Created-by: chenxingyu18 Commit-by: chen-xing-yu Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动，包括改动的原因和所采取的方法。--> 将 index/、pooling/、conv/ 下所有 arch32 目录与引用统一改为 arch22，与全仓主流命名约定对齐（arch22 用于 ascend910b/910_93，arch35 用于 ascend950）。 - 目录重命名 arch32 → arch22（op_host/op_tiling、op_kernel、tests/ut 等） - 文件内容：include 路径、CMakeLists 的 SUPPORT_TILING_DIR、注释中的 arch32 → arch22 - arch35 目录与引用未受影响 - 顺带修复两个被禁用（main() 提前 return 0）的 eager example：index_add、scatter_nd_update ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。例如：关联Issue #000--> #4493 ## 测试 <!--描述进行了哪些测试来验证你的改动。--> 基于 ascend910b（CANN 9.1.0 / 910B3）完成统一打包构建 + eager 验证，涉及算子 index_fill、scatter_nd_update、masked_scatter、adaptive_max_pool3d(_grad)、max_pool3d_grad_with_argmax、avg_pool3_d、inplace_index_add_with_sorted、conv3d_backprop_filter_v2、conv3d_backprop_input_v2、conv3d_transpose_v2 均验证通过；并用改名测试（移除 vendor kernel 后 example 失败）确认跑的是改名后的实现而非 stock CANN。 ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [x] 其他，请描述：命名统一（arch32 → arch22） ## AI/Agent生成声明 <!-- [x] 表示选中 --> - [x] AI辅助编写 See merge request: cann/ops-nn!8210	1 个月前
tests	refactor: 统一 arch32 → arch22 命名（index/pooling/conv） Co-authored-by: chen-xing-yu<chenxingyu18@huawei.com> # message auto-generated for no-merge-commit merge: !8210 merge refactor/arch32_to_arch22 into master refactor: 统一 arch32 → arch22 命名（index/pooling/conv） Created-by: chenxingyu18 Commit-by: chen-xing-yu Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动，包括改动的原因和所采取的方法。--> 将 index/、pooling/、conv/ 下所有 arch32 目录与引用统一改为 arch22，与全仓主流命名约定对齐（arch22 用于 ascend910b/910_93，arch35 用于 ascend950）。 - 目录重命名 arch32 → arch22（op_host/op_tiling、op_kernel、tests/ut 等） - 文件内容：include 路径、CMakeLists 的 SUPPORT_TILING_DIR、注释中的 arch32 → arch22 - arch35 目录与引用未受影响 - 顺带修复两个被禁用（main() 提前 return 0）的 eager example：index_add、scatter_nd_update ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。例如：关联Issue #000--> #4493 ## 测试 <!--描述进行了哪些测试来验证你的改动。--> 基于 ascend910b（CANN 9.1.0 / 910B3）完成统一打包构建 + eager 验证，涉及算子 index_fill、scatter_nd_update、masked_scatter、adaptive_max_pool3d(_grad)、max_pool3d_grad_with_argmax、avg_pool3_d、inplace_index_add_with_sorted、conv3d_backprop_filter_v2、conv3d_backprop_input_v2、conv3d_transpose_v2 均验证通过；并用改名测试（移除 vendor kernel 后 example 失败）确认跑的是改名后的实现而非 stock CANN。 ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [x] 其他，请描述：命名统一（arch32 → arch22） ## AI/Agent生成声明 <!-- [x] 表示选中 --> - [x] AI辅助编写 See merge request: cann/ops-nn!8210	1 个月前
CMakeLists.txt	refactor: 统一 arch32 → arch22 命名（index/pooling/conv） Co-authored-by: chen-xing-yu<chenxingyu18@huawei.com> # message auto-generated for no-merge-commit merge: !8210 merge refactor/arch32_to_arch22 into master refactor: 统一 arch32 → arch22 命名（index/pooling/conv） Created-by: chenxingyu18 Commit-by: chen-xing-yu Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动，包括改动的原因和所采取的方法。--> 将 index/、pooling/、conv/ 下所有 arch32 目录与引用统一改为 arch22，与全仓主流命名约定对齐（arch22 用于 ascend910b/910_93，arch35 用于 ascend950）。 - 目录重命名 arch32 → arch22（op_host/op_tiling、op_kernel、tests/ut 等） - 文件内容：include 路径、CMakeLists 的 SUPPORT_TILING_DIR、注释中的 arch32 → arch22 - arch35 目录与引用未受影响 - 顺带修复两个被禁用（main() 提前 return 0）的 eager example：index_add、scatter_nd_update ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。例如：关联Issue #000--> #4493 ## 测试 <!--描述进行了哪些测试来验证你的改动。--> 基于 ascend910b（CANN 9.1.0 / 910B3）完成统一打包构建 + eager 验证，涉及算子 index_fill、scatter_nd_update、masked_scatter、adaptive_max_pool3d(_grad)、max_pool3d_grad_with_argmax、avg_pool3_d、inplace_index_add_with_sorted、conv3d_backprop_filter_v2、conv3d_backprop_input_v2、conv3d_transpose_v2 均验证通过；并用改名测试（移除 vendor kernel 后 example 失败）确认跑的是改名后的实现而非 stock CANN。 ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [x] 其他，请描述：命名统一（arch32 → arch22） ## AI/Agent生成声明 <!-- [x] 表示选中 --> - [x] AI辅助编写 See merge request: cann/ops-nn!8210	1 个月前
README.md	批量修改md中冗余空格 Co-authored-by: wuyao51511<wuyao61@h-partners.com> # message auto-generated for no-merge-commit merge: !6027 merge master into master 批量修改md中冗余空格 Created-by: wuyao51511 Commit-by: wuyao51511 Merged-by: cann-robot Description: ## 描述文档空格问题清理 ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。例如：关联Issue #000--> ## 测试 <!--描述进行了哪些测试来验证你的改动。--> ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-nn!6027	2 个月前

ScatterNdUpdate

产品支持情况

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

功能说明

算子功能：根据indices在给定变量内，将updates应用于变量的单个值或切片。
计算公式：ref[indices[k]]=updates[k]
- ref是一个维度为P的张量Tensor；
- indices是一个维度为Q的整型张量Tensor；
- indices的shape一定是[d0,..., d_Q-2, K]，此处0<K<=P；
- updates是一个维度为Q−1+P−K的张量：[d_0, ..., d_{Q-2}, ref.shape[K], ..., ref.shape[P-1]]

参数说明

参数名	输入/输出/属性	描述	数据类型	数据格式
var	输入	表示一个待被更新的张量,等同于公式中的`ref`	DT_INT8、DT_INT64、DT_BOOL、DT_FLOAT16、DT_BF16、DT_FLOAT、FLOAT8_E4M3FN、FLOAT8_E5M2、HIFLOAT8	ND
indices	输入	一个索引张量，索引到公式中的`ref`	INT32、INT64。	ND
updates	输入	使用此张量来更新var张量，必须跟输入var张量保持一样的数据类型	DT_INT8、DT_INT64、DT_BOOL、DT_FLOAT16、DT_BF16、DT_FLOAT、FLOAT8_E4M3FN、FLOAT8_E5M2、HIFLOAT8	ND
var	输出	表示更新后的张量	DT_INT8、DT_INT64、DT_BOOL、DT_FLOAT16、DT_BF16、DT_FLOAT、FLOAT8_E4M3FN、FLOAT8_E5M2、HIFLOAT8	ND
use_locking	可选属性	可选属性，默认值为`false`,如果为`true`,这次操作将会被一个lock保护	BOOL	-

Atlas A3 训练系列产品/Atlas A3 推理系列产品、Atlas A2 训练系列产品/Atlas A2 推理系列产品：FLOAT8_E4M3FN、FLOAT8_E5M2、HIFLOAT8数据类型在该型号中不支持。

约束说明

输入shape限制：
- indices至少是2维，其最后1维的大小不能超过varRef的维度大小。
- 假设indices最后1维的大小是a，则updates的shape等于indices除最后1维外的shape加上varRef除前a维外的shape。举例：varRef的shape是(4, 5, 6)，indices的shape是(3, 2)，则updates的shape必须是(3, 6)。

调用说明

调用方式	样例代码	说明
aclnn接口	test_aclnn_scatter_nd_update	通过aclnnScatterNdUpdate接口方式调用ScatterNdUpdate算子。