| 文件 | 最后提交记录 | 最后更新时间 |
|---|---|---|
index clean code Co-authored-by: yue-ma<mayue54@huawei.com> # message auto-generated for no-merge-commit merge: !9028 merge master_bak into master index clean code Created-by: yue-ma Commit-by: yue-ma Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动，包括改动的原因和所采取的方法。--> index目录下进行clean code整改 ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。例如：关联Issue #000--> https://gitcode.com/cann/ops-nn/issues/4233 ## 测试 <!--描述进行了哪些测试来验证你的改动。--> UT/ST通过 ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> NA ## 类型标签 <!-- [x] 表示选中 --> - [x] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： ## AI/Agent生成声明 <!-- [x] 表示选中 --> - [ ] AI辅助编写 See merge request: cann/ops-nn!9028 | 2 个月前 | |
legacy下线：整改NN算子图模式注册与日志规范 Co-authored-by: zhouxuan78<zhouxuan78@huawei.com> # message auto-generated for no-merge-commit merge: !8388 merge master into master legacy下线：整改NN算子图模式注册与日志规范 Created-by: zhouxuan78 Commit-by: zhouxuan78 Merged-by: cann-robot Description: ## 描述 本 PR 用于 ops-nn 仓 Legacy 下线整改，补齐部分算子的直调/图模式注册、InferShape / InferDataType 拆分、日志规范整改及必要的编译/预提交问题修复。 1. 补齐多个算子的 op_graph 侧 InferDataType / graph infer 注册文件，并同步补充对应 CMakeLists.txt： - 如 FastGeluV2、HardShrinkGrad、HardSigmoidGrad、HardSwish、HardSwishGrad、Selu、SeluGrad、Shrink、SoftplusGrad、SoftplusV2、SoftplusV2Grad、Softsign、SoftsignGrad、SmoothL1Loss、SoftMarginLoss、LpNormUpdateV2、ApplyFtrlV2、ApplyMomentum、ApplyRMSProp 等。 2. 调整部分算子的 op_host infer 文件： - 将 InferDataType 逻辑从 infershape 文件中拆分/迁移到 op_graph 侧； - 保留 InferShape 逻辑，避免 shape/type 混在同一处注册； - 收敛 IndexCheck infershape 日志，仅保留入口日志。 3. 补充 SoftMarginLoss 图模式原型： - 新增 loss/soft_margin_loss/op_graph/soft_margin_loss_proto.h； - 新增 OPS_PROTO_DEF_SOFTMARGINLOSS 去重宏，避免重复注册； - 补充 SoftMarginLoss 的 op_graph/CMakeLists.txt 和 graph infer 注册。 4. 日志整改： - 为 Arch35 tiling 入口补充统一入口日志； - 对本 PR 修改范围内新增/调整的外部输入错误日志，按规范使用 OP_LOGE_FOR_INVALID_* 系列上报接口； - 收敛部分过多的 tiling / infershape 日志打印，避免循环或冗余打印。 5. 回退/收敛不属于本次整改范围的 Softshrink 重命名相关改动： - Softshrink -> SoftShrink 的命名调整不放在本 PR 中，后续单独 PR 处理。 6. 修复 pre-commit 格式问题： - 按仓库 .clang-format 修复格式； - 不修改 .pre-commit-config.yaml。 影响范围 主要影响 activation、index、loss、norm、optim、quant、vfusion 等目录下 Legacy 下线相关算子的 op_graph 注册、InferDataType 拆分、tiling 入口日志和错误日志规范。 ## 变更内容 ## 关联的Issue ## 测试 ## 文档更新 ## 类型标签 - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [x] 其他：Legacy包下线整改 See merge request: cann/ops-nn!8388 | 1 个月前 | |
legacy下线：整改NN算子图模式注册与日志规范 Co-authored-by: zhouxuan78<zhouxuan78@huawei.com> # message auto-generated for no-merge-commit merge: !8388 merge master into master legacy下线：整改NN算子图模式注册与日志规范 Created-by: zhouxuan78 Commit-by: zhouxuan78 Merged-by: cann-robot Description: ## 描述 本 PR 用于 ops-nn 仓 Legacy 下线整改，补齐部分算子的直调/图模式注册、InferShape / InferDataType 拆分、日志规范整改及必要的编译/预提交问题修复。 1. 补齐多个算子的 op_graph 侧 InferDataType / graph infer 注册文件，并同步补充对应 CMakeLists.txt： - 如 FastGeluV2、HardShrinkGrad、HardSigmoidGrad、HardSwish、HardSwishGrad、Selu、SeluGrad、Shrink、SoftplusGrad、SoftplusV2、SoftplusV2Grad、Softsign、SoftsignGrad、SmoothL1Loss、SoftMarginLoss、LpNormUpdateV2、ApplyFtrlV2、ApplyMomentum、ApplyRMSProp 等。 2. 调整部分算子的 op_host infer 文件： - 将 InferDataType 逻辑从 infershape 文件中拆分/迁移到 op_graph 侧； - 保留 InferShape 逻辑，避免 shape/type 混在同一处注册； - 收敛 IndexCheck infershape 日志，仅保留入口日志。 3. 补充 SoftMarginLoss 图模式原型： - 新增 loss/soft_margin_loss/op_graph/soft_margin_loss_proto.h； - 新增 OPS_PROTO_DEF_SOFTMARGINLOSS 去重宏，避免重复注册； - 补充 SoftMarginLoss 的 op_graph/CMakeLists.txt 和 graph infer 注册。 4. 日志整改： - 为 Arch35 tiling 入口补充统一入口日志； - 对本 PR 修改范围内新增/调整的外部输入错误日志，按规范使用 OP_LOGE_FOR_INVALID_* 系列上报接口； - 收敛部分过多的 tiling / infershape 日志打印，避免循环或冗余打印。 5. 回退/收敛不属于本次整改范围的 Softshrink 重命名相关改动： - Softshrink -> SoftShrink 的命名调整不放在本 PR 中，后续单独 PR 处理。 6. 修复 pre-commit 格式问题： - 按仓库 .clang-format 修复格式； - 不修改 .pre-commit-config.yaml。 影响范围 主要影响 activation、index、loss、norm、optim、quant、vfusion 等目录下 Legacy 下线相关算子的 op_graph 注册、InferDataType 拆分、tiling 入口日志和错误日志规范。 ## 变更内容 ## 关联的Issue ## 测试 ## 文档更新 ## 类型标签 - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [x] 其他：Legacy包下线整改 See merge request: cann/ops-nn!8388 | 1 个月前 | |
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 | 3 个月前 | |
修改index、index_put_v2算子aclnn接口L2调用点，增加IndexCheck越界检查 Co-authored-by: BoyangZhang<zhangboyang15@h-partners.com> # message auto-generated for no-merge-commit merge: !4957 merge index_check_l2_callpoint into master 修改index、index_put_v2算子aclnn接口L2调用点，增加IndexCheck越界检查 Created-by: BoyangZhang Commit-by: BoyangZhang Merged-by: cann-robot Description: ## 描述 修改index、index_put_v2算子aclnn接口L2调用点，增加IndexCheck越界检查 <!--在这里详细描述你的改动，包括改动的原因和所采取的方法。--> ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。例如：关联Issue #000--> https://gitcode.com/cann/ops-nn/issues/2750 ## 测试 <!--描述进行了哪些测试来验证你的改动。--> ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-nn!4957 | 5 个月前 | |
indexCheck算子适配950 Co-authored-by: kangjiaming<1159380836@qq.com> # message auto-generated for no-merge-commit merge: !8386 merge index_check into master indexCheck算子适配950 Created-by: kangjiaming Commit-by: kangjiaming Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动，包括改动的原因和所采取的方法。--> 本次 PR 为 IndexCheck 算子适配 ASCEND950 平台，并新增多项输入合法性校验，提升算子在 950 平台的运行时安全性与鲁棒性。 ### 主要改动 - **op_api/index_check.cpp — 增加 ASCEND950 平台支持**：在 IndexCheck 入口的平台判断逻辑中新增 SocVersion::ASCEND950，使算子可以在 950 平台上继续执行，同时更新日志描述。 - **op_api/index_check.cpp — 新增 bounds 数据类型校验**：新增 bounds->GetDataType() 必须为 DT_INT64 的检查，不满足时跳过 IndexCheck 并输出警告。 - **op_api/index_check.cpp — 新增 indices 张量 dtype 一致性与数量上限校验**：IsAiCoreSupport 现在检查所有 indices 张量的 dtype 是否一致，不一致则返回 false；IndexCheck 中新增 indices 数量超过 8 时的跳过逻辑。 - **op_host/index_check_tiling.cpp — 新增 tiling 层张量数量与维度数硬校验**：在 IndexCheckTiling::Init 中新增 tensorId_ > MAX_TENSOR_NUM 和单张量维度数 > MAX_DIM_NUM（8）的错误检查，不满足时直接返回 GRAPH_FAILED。 - **op_host/index_check_def.cpp — 注册 ascend950 AICore 配置**：在 AICore 配置列表中添加 "ascend950"，使算子可在新平台上加载对应的 tiling/kernel 配置。 - **README.md**：更新 Ascend 950PR/950DT 支持状态由 × 改为 √。 ### 改动原因 IndexCheck 作为 aclnnIndex、aclnnIndexPutV2 等索引类 aclnn 算子的公共前置检查算子（L0 内部算子），此前仅支持 910B 和 910_93 平台。本次适配使 950 平台上的索引类算子具备越界检查能力，同时通过增强输入校验提前拦截非法输入，避免后续 kernel 执行异常。 ## 关联的Issue 关联 Issue：[Requirement|需求建议]: IndexCheck 算子适配 Ascend 950 平台并增强输入校验 ## 测试 - CI 编译流水线全部通过（X86/ARM/950 等多个编译任务 SUCCESS）。 - UT 测试全部通过（ophost/opapi/kernel/opgraph）。 - PreSmoke 测试通过。 - API_Check 通过。 - 代码检视意见已修复（%d → %zu 格式说明符、移除多余 static_cast）。 - codecheck 格式问题已修复（clang-format、end-of-file-fixer）。 ## 文档更新 - 更新了 index/index_check/README.md，将 Ascend 950PR/950DT 的支持状态由 × 改为 √。 ## 类型标签 - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [x] 文档更新 - [ ] 其他，请描述： ## AI/Agent生成声明 - [x] AI辅助编写 ## 关联的Issue https://gitcode.com/cann/ops-nn/issues/4694 ## 测试 <!--描述进行了哪些测试来验证你的改动。--> ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： ## AI/Agent生成声明 <!-- [x] 表示选中 --> - [ ] AI辅助编写 See merge request: cann/ops-nn!8386 | 2 个月前 |
IndexCheck
产品支持情况
| 产品 | 是否支持 |
|---|---|
| Ascend 950PR/Ascend 950DT | √ |
| Atlas A3 训练系列产品/Atlas A3 推理系列产品 | √ |
| Atlas A2 训练系列产品/Atlas A2 推理系列产品 | √ |
| Atlas 200I/500 A2 推理产品 | × |
| Atlas 推理系列产品 | × |
| Atlas 训练系列产品 | × |
功能说明
- 算子功能：检查索引是否越界。当索引超出边界值范围时，算子会通过assert报错终止程序。
- 应用场景：用于索引类aclnn l2接口，在执行索引操作前进行索引合法性检查。
参数说明
| 参数名 | 输入/输出/属性 | 描述 | 数据类型 | 数据格式 |
|---|---|---|---|---|
| bounds | 输入 | 各维度的边界值（即各维度的size），长度需与indices列表中tensor个数一致。 | INT64 | - |
| indices | 输入 | 索引张量列表，列表中每个张量按顺序对应bounds中的一个边界值。 | INT32、INT64 | ND |
约束说明
- bounds的长度不能超过8。
- bounds的长度需要等于indices列表中tensor的个数。
- indices中的索引值支持负数。
- 若索引越界，算子会通过assert报错终止程序。