| 文件 | 最后提交记录 | 最后更新时间 |
|---|---|---|
Migrate index AICPU shape operators Co-authored-by: jialimin1<jialimin1@huawei.com> # message auto-generated for no-merge-commit merge: !7519 merge migrate-index-to-addr-aicpu into master Migrate index AICPU shape operators Created-by: jialimin1 Commit-by: jialimin1 Merged-by: cann-robot Description: ## 描述 将三个 AICPU shape 类算子从源仓迁移至 ops-nn 仓库，统一算子仓管理，共新增 43 个文件，并同步更新算子清单 docs/zh/op_list.md： 1. **IndexToAddr**：根据块索引、原始矩阵 shape、块 shape 和基地址信息，生成块内每一行对应的地址表。输入 base_addr（shape=[2]）与块索引 x（shape=[2]），输出地址表 shape 为 [block_size[0], 4]，每行 [rank_id, base_addr[0]+bias_addr, base_addr[1]+bias_addr, row_size*dtype_size]，支持 INT64/UINT64。 2. **NonZeroWithValueShape**：根据 NonZeroWithValue 的 count 输入更新 value/index 输出 shape，分别置为 [count[0]] 与 [2, count[0]]。 3. **NonZeroWithValueShapeV2**：V2 版本，在 V1 基础上新增 GetInputTypeAndCheck 显式校验 index/count 必须为 INT32，输出语义与 V1 一致。 每个算子按 ops-nn 标准目录结构组织，包含 op_graph（IR 原型）、op_host（infershape）、op_kernel_aicpu（kernel 实现与 REGISTER_CPU_KERNEL 注册）、examples（图模式调用样例）、tests/ut（单元测试）、各级 CMakeLists.txt 与 README.md。 ## 关联的Issue #4163 ## 测试 根据代码变更，测试场景如下： 1. **编译构建测试** - 对三个算子（IndexToAddr / NonZeroWithValueShape / NonZeroWithValueShapeV2）分别执行各级 CMakeLists.txt 配置与编译，验证新增 CMake 配置变更正确，算子可正常构建。 2. **算子加载/注册验证** - 验证 REGISTER_CPU_KERNEL 注册成功，三个算子可被框架正确识别与加载。 3. **InferShape 验证** - 验证 op_host/*_infershape.cpp 输出 shape 推导：IndexToAddr 输出 [block_size[0], 4]；NonZeroWithValueShape[V2] 输出 out_value=[count]、out_index=[2, count]。 4. **单元测试** - 运行 tests/ut/op_kernel_aicpu/test_*.cpp，覆盖 INT64/UINT64（IndexToAddr）、INT32 校验（NonZeroWithValueShapeV2）等数据类型分支与参数校验逻辑。 5. **图模式接口调用验证** - 运行 examples/test_geir_*.cpp，通过算子 IR 构图方式调用三个算子，验证端到端功能与输出正确性。 6. **参数校验测试** - 覆盖 Check() 中的边界场景：base_addr/x 数据类型一致且 shape=[2]、ori_shape/block_size 为 2 元素、block_size[0]*4 <= output.NumElements()、storage mode 仅支持 Matrix、V2 版本 index/count 必须为 DT_INT32、count 至少 1 个元素等。 7. **算子清单登记验证** - 确认 docs/zh/op_list.md 中三个算子条目正确显示，分类为 index，类型为 AI CPU，且 README 链接可达。 ## 文档更新 - 新增 index/index_to_addr/README.md、index/non_zero_with_value_shape/README.md、index/non_zero_with_value_shape_v2/README.md：产品支持情况、功能说明、参数说明、约束说明与调用说明。 - 更新 docs/zh/op_list.md：登记 IndexToAddr、NonZeroWithValueShape、NonZeroWithValueShapeV2 三个算子条目。 ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： ## AI/Agent生成声明 <!-- [x] 表示选中 --> - [ ] AI辅助编写 See merge request: cann/ops-nn!7519 | 1 个月前 | |
Migrate index AICPU shape operators Co-authored-by: jialimin1<jialimin1@huawei.com> # message auto-generated for no-merge-commit merge: !7519 merge migrate-index-to-addr-aicpu into master Migrate index AICPU shape operators Created-by: jialimin1 Commit-by: jialimin1 Merged-by: cann-robot Description: ## 描述 将三个 AICPU shape 类算子从源仓迁移至 ops-nn 仓库，统一算子仓管理，共新增 43 个文件，并同步更新算子清单 docs/zh/op_list.md： 1. **IndexToAddr**：根据块索引、原始矩阵 shape、块 shape 和基地址信息，生成块内每一行对应的地址表。输入 base_addr（shape=[2]）与块索引 x（shape=[2]），输出地址表 shape 为 [block_size[0], 4]，每行 [rank_id, base_addr[0]+bias_addr, base_addr[1]+bias_addr, row_size*dtype_size]，支持 INT64/UINT64。 2. **NonZeroWithValueShape**：根据 NonZeroWithValue 的 count 输入更新 value/index 输出 shape，分别置为 [count[0]] 与 [2, count[0]]。 3. **NonZeroWithValueShapeV2**：V2 版本，在 V1 基础上新增 GetInputTypeAndCheck 显式校验 index/count 必须为 INT32，输出语义与 V1 一致。 每个算子按 ops-nn 标准目录结构组织，包含 op_graph（IR 原型）、op_host（infershape）、op_kernel_aicpu（kernel 实现与 REGISTER_CPU_KERNEL 注册）、examples（图模式调用样例）、tests/ut（单元测试）、各级 CMakeLists.txt 与 README.md。 ## 关联的Issue #4163 ## 测试 根据代码变更，测试场景如下： 1. **编译构建测试** - 对三个算子（IndexToAddr / NonZeroWithValueShape / NonZeroWithValueShapeV2）分别执行各级 CMakeLists.txt 配置与编译，验证新增 CMake 配置变更正确，算子可正常构建。 2. **算子加载/注册验证** - 验证 REGISTER_CPU_KERNEL 注册成功，三个算子可被框架正确识别与加载。 3. **InferShape 验证** - 验证 op_host/*_infershape.cpp 输出 shape 推导：IndexToAddr 输出 [block_size[0], 4]；NonZeroWithValueShape[V2] 输出 out_value=[count]、out_index=[2, count]。 4. **单元测试** - 运行 tests/ut/op_kernel_aicpu/test_*.cpp，覆盖 INT64/UINT64（IndexToAddr）、INT32 校验（NonZeroWithValueShapeV2）等数据类型分支与参数校验逻辑。 5. **图模式接口调用验证** - 运行 examples/test_geir_*.cpp，通过算子 IR 构图方式调用三个算子，验证端到端功能与输出正确性。 6. **参数校验测试** - 覆盖 Check() 中的边界场景：base_addr/x 数据类型一致且 shape=[2]、ori_shape/block_size 为 2 元素、block_size[0]*4 <= output.NumElements()、storage mode 仅支持 Matrix、V2 版本 index/count 必须为 DT_INT32、count 至少 1 个元素等。 7. **算子清单登记验证** - 确认 docs/zh/op_list.md 中三个算子条目正确显示，分类为 index，类型为 AI CPU，且 README 链接可达。 ## 文档更新 - 新增 index/index_to_addr/README.md、index/non_zero_with_value_shape/README.md、index/non_zero_with_value_shape_v2/README.md：产品支持情况、功能说明、参数说明、约束说明与调用说明。 - 更新 docs/zh/op_list.md：登记 IndexToAddr、NonZeroWithValueShape、NonZeroWithValueShapeV2 三个算子条目。 ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： ## AI/Agent生成声明 <!-- [x] 表示选中 --> - [ ] AI辅助编写 See merge request: cann/ops-nn!7519 | 1 个月前 | |
Migrate index AICPU shape operators Co-authored-by: jialimin1<jialimin1@huawei.com> # message auto-generated for no-merge-commit merge: !7519 merge migrate-index-to-addr-aicpu into master Migrate index AICPU shape operators Created-by: jialimin1 Commit-by: jialimin1 Merged-by: cann-robot Description: ## 描述 将三个 AICPU shape 类算子从源仓迁移至 ops-nn 仓库，统一算子仓管理，共新增 43 个文件，并同步更新算子清单 docs/zh/op_list.md： 1. **IndexToAddr**：根据块索引、原始矩阵 shape、块 shape 和基地址信息，生成块内每一行对应的地址表。输入 base_addr（shape=[2]）与块索引 x（shape=[2]），输出地址表 shape 为 [block_size[0], 4]，每行 [rank_id, base_addr[0]+bias_addr, base_addr[1]+bias_addr, row_size*dtype_size]，支持 INT64/UINT64。 2. **NonZeroWithValueShape**：根据 NonZeroWithValue 的 count 输入更新 value/index 输出 shape，分别置为 [count[0]] 与 [2, count[0]]。 3. **NonZeroWithValueShapeV2**：V2 版本，在 V1 基础上新增 GetInputTypeAndCheck 显式校验 index/count 必须为 INT32，输出语义与 V1 一致。 每个算子按 ops-nn 标准目录结构组织，包含 op_graph（IR 原型）、op_host（infershape）、op_kernel_aicpu（kernel 实现与 REGISTER_CPU_KERNEL 注册）、examples（图模式调用样例）、tests/ut（单元测试）、各级 CMakeLists.txt 与 README.md。 ## 关联的Issue #4163 ## 测试 根据代码变更，测试场景如下： 1. **编译构建测试** - 对三个算子（IndexToAddr / NonZeroWithValueShape / NonZeroWithValueShapeV2）分别执行各级 CMakeLists.txt 配置与编译，验证新增 CMake 配置变更正确，算子可正常构建。 2. **算子加载/注册验证** - 验证 REGISTER_CPU_KERNEL 注册成功，三个算子可被框架正确识别与加载。 3. **InferShape 验证** - 验证 op_host/*_infershape.cpp 输出 shape 推导：IndexToAddr 输出 [block_size[0], 4]；NonZeroWithValueShape[V2] 输出 out_value=[count]、out_index=[2, count]。 4. **单元测试** - 运行 tests/ut/op_kernel_aicpu/test_*.cpp，覆盖 INT64/UINT64（IndexToAddr）、INT32 校验（NonZeroWithValueShapeV2）等数据类型分支与参数校验逻辑。 5. **图模式接口调用验证** - 运行 examples/test_geir_*.cpp，通过算子 IR 构图方式调用三个算子，验证端到端功能与输出正确性。 6. **参数校验测试** - 覆盖 Check() 中的边界场景：base_addr/x 数据类型一致且 shape=[2]、ori_shape/block_size 为 2 元素、block_size[0]*4 <= output.NumElements()、storage mode 仅支持 Matrix、V2 版本 index/count 必须为 DT_INT32、count 至少 1 个元素等。 7. **算子清单登记验证** - 确认 docs/zh/op_list.md 中三个算子条目正确显示，分类为 index，类型为 AI CPU，且 README 链接可达。 ## 文档更新 - 新增 index/index_to_addr/README.md、index/non_zero_with_value_shape/README.md、index/non_zero_with_value_shape_v2/README.md：产品支持情况、功能说明、参数说明、约束说明与调用说明。 - 更新 docs/zh/op_list.md：登记 IndexToAddr、NonZeroWithValueShape、NonZeroWithValueShapeV2 三个算子条目。 ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： ## AI/Agent生成声明 <!-- [x] 表示选中 --> - [ ] AI辅助编写 See merge request: cann/ops-nn!7519 | 1 个月前 | |
Migrate index AICPU shape operators Co-authored-by: jialimin1<jialimin1@huawei.com> # message auto-generated for no-merge-commit merge: !7519 merge migrate-index-to-addr-aicpu into master Migrate index AICPU shape operators Created-by: jialimin1 Commit-by: jialimin1 Merged-by: cann-robot Description: ## 描述 将三个 AICPU shape 类算子从源仓迁移至 ops-nn 仓库，统一算子仓管理，共新增 43 个文件，并同步更新算子清单 docs/zh/op_list.md： 1. **IndexToAddr**：根据块索引、原始矩阵 shape、块 shape 和基地址信息，生成块内每一行对应的地址表。输入 base_addr（shape=[2]）与块索引 x（shape=[2]），输出地址表 shape 为 [block_size[0], 4]，每行 [rank_id, base_addr[0]+bias_addr, base_addr[1]+bias_addr, row_size*dtype_size]，支持 INT64/UINT64。 2. **NonZeroWithValueShape**：根据 NonZeroWithValue 的 count 输入更新 value/index 输出 shape，分别置为 [count[0]] 与 [2, count[0]]。 3. **NonZeroWithValueShapeV2**：V2 版本，在 V1 基础上新增 GetInputTypeAndCheck 显式校验 index/count 必须为 INT32，输出语义与 V1 一致。 每个算子按 ops-nn 标准目录结构组织，包含 op_graph（IR 原型）、op_host（infershape）、op_kernel_aicpu（kernel 实现与 REGISTER_CPU_KERNEL 注册）、examples（图模式调用样例）、tests/ut（单元测试）、各级 CMakeLists.txt 与 README.md。 ## 关联的Issue #4163 ## 测试 根据代码变更，测试场景如下： 1. **编译构建测试** - 对三个算子（IndexToAddr / NonZeroWithValueShape / NonZeroWithValueShapeV2）分别执行各级 CMakeLists.txt 配置与编译，验证新增 CMake 配置变更正确，算子可正常构建。 2. **算子加载/注册验证** - 验证 REGISTER_CPU_KERNEL 注册成功，三个算子可被框架正确识别与加载。 3. **InferShape 验证** - 验证 op_host/*_infershape.cpp 输出 shape 推导：IndexToAddr 输出 [block_size[0], 4]；NonZeroWithValueShape[V2] 输出 out_value=[count]、out_index=[2, count]。 4. **单元测试** - 运行 tests/ut/op_kernel_aicpu/test_*.cpp，覆盖 INT64/UINT64（IndexToAddr）、INT32 校验（NonZeroWithValueShapeV2）等数据类型分支与参数校验逻辑。 5. **图模式接口调用验证** - 运行 examples/test_geir_*.cpp，通过算子 IR 构图方式调用三个算子，验证端到端功能与输出正确性。 6. **参数校验测试** - 覆盖 Check() 中的边界场景：base_addr/x 数据类型一致且 shape=[2]、ori_shape/block_size 为 2 元素、block_size[0]*4 <= output.NumElements()、storage mode 仅支持 Matrix、V2 版本 index/count 必须为 DT_INT32、count 至少 1 个元素等。 7. **算子清单登记验证** - 确认 docs/zh/op_list.md 中三个算子条目正确显示，分类为 index，类型为 AI CPU，且 README 链接可达。 ## 文档更新 - 新增 index/index_to_addr/README.md、index/non_zero_with_value_shape/README.md、index/non_zero_with_value_shape_v2/README.md：产品支持情况、功能说明、参数说明、约束说明与调用说明。 - 更新 docs/zh/op_list.md：登记 IndexToAddr、NonZeroWithValueShape、NonZeroWithValueShapeV2 三个算子条目。 ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： ## AI/Agent生成声明 <!-- [x] 表示选中 --> - [ ] AI辅助编写 See merge request: cann/ops-nn!7519 | 1 个月前 | |
Migrate index AICPU shape operators Co-authored-by: jialimin1<jialimin1@huawei.com> # message auto-generated for no-merge-commit merge: !7519 merge migrate-index-to-addr-aicpu into master Migrate index AICPU shape operators Created-by: jialimin1 Commit-by: jialimin1 Merged-by: cann-robot Description: ## 描述 将三个 AICPU shape 类算子从源仓迁移至 ops-nn 仓库，统一算子仓管理，共新增 43 个文件，并同步更新算子清单 docs/zh/op_list.md： 1. **IndexToAddr**：根据块索引、原始矩阵 shape、块 shape 和基地址信息，生成块内每一行对应的地址表。输入 base_addr（shape=[2]）与块索引 x（shape=[2]），输出地址表 shape 为 [block_size[0], 4]，每行 [rank_id, base_addr[0]+bias_addr, base_addr[1]+bias_addr, row_size*dtype_size]，支持 INT64/UINT64。 2. **NonZeroWithValueShape**：根据 NonZeroWithValue 的 count 输入更新 value/index 输出 shape，分别置为 [count[0]] 与 [2, count[0]]。 3. **NonZeroWithValueShapeV2**：V2 版本，在 V1 基础上新增 GetInputTypeAndCheck 显式校验 index/count 必须为 INT32，输出语义与 V1 一致。 每个算子按 ops-nn 标准目录结构组织，包含 op_graph（IR 原型）、op_host（infershape）、op_kernel_aicpu（kernel 实现与 REGISTER_CPU_KERNEL 注册）、examples（图模式调用样例）、tests/ut（单元测试）、各级 CMakeLists.txt 与 README.md。 ## 关联的Issue #4163 ## 测试 根据代码变更，测试场景如下： 1. **编译构建测试** - 对三个算子（IndexToAddr / NonZeroWithValueShape / NonZeroWithValueShapeV2）分别执行各级 CMakeLists.txt 配置与编译，验证新增 CMake 配置变更正确，算子可正常构建。 2. **算子加载/注册验证** - 验证 REGISTER_CPU_KERNEL 注册成功，三个算子可被框架正确识别与加载。 3. **InferShape 验证** - 验证 op_host/*_infershape.cpp 输出 shape 推导：IndexToAddr 输出 [block_size[0], 4]；NonZeroWithValueShape[V2] 输出 out_value=[count]、out_index=[2, count]。 4. **单元测试** - 运行 tests/ut/op_kernel_aicpu/test_*.cpp，覆盖 INT64/UINT64（IndexToAddr）、INT32 校验（NonZeroWithValueShapeV2）等数据类型分支与参数校验逻辑。 5. **图模式接口调用验证** - 运行 examples/test_geir_*.cpp，通过算子 IR 构图方式调用三个算子，验证端到端功能与输出正确性。 6. **参数校验测试** - 覆盖 Check() 中的边界场景：base_addr/x 数据类型一致且 shape=[2]、ori_shape/block_size 为 2 元素、block_size[0]*4 <= output.NumElements()、storage mode 仅支持 Matrix、V2 版本 index/count 必须为 DT_INT32、count 至少 1 个元素等。 7. **算子清单登记验证** - 确认 docs/zh/op_list.md 中三个算子条目正确显示，分类为 index，类型为 AI CPU，且 README 链接可达。 ## 文档更新 - 新增 index/index_to_addr/README.md、index/non_zero_with_value_shape/README.md、index/non_zero_with_value_shape_v2/README.md：产品支持情况、功能说明、参数说明、约束说明与调用说明。 - 更新 docs/zh/op_list.md：登记 IndexToAddr、NonZeroWithValueShape、NonZeroWithValueShapeV2 三个算子条目。 ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： ## AI/Agent生成声明 <!-- [x] 表示选中 --> - [ ] AI辅助编写 See merge request: cann/ops-nn!7519 | 1 个月前 | |
Migrate index AICPU shape operators Co-authored-by: jialimin1<jialimin1@huawei.com> # message auto-generated for no-merge-commit merge: !7519 merge migrate-index-to-addr-aicpu into master Migrate index AICPU shape operators Created-by: jialimin1 Commit-by: jialimin1 Merged-by: cann-robot Description: ## 描述 将三个 AICPU shape 类算子从源仓迁移至 ops-nn 仓库，统一算子仓管理，共新增 43 个文件，并同步更新算子清单 docs/zh/op_list.md： 1. **IndexToAddr**：根据块索引、原始矩阵 shape、块 shape 和基地址信息，生成块内每一行对应的地址表。输入 base_addr（shape=[2]）与块索引 x（shape=[2]），输出地址表 shape 为 [block_size[0], 4]，每行 [rank_id, base_addr[0]+bias_addr, base_addr[1]+bias_addr, row_size*dtype_size]，支持 INT64/UINT64。 2. **NonZeroWithValueShape**：根据 NonZeroWithValue 的 count 输入更新 value/index 输出 shape，分别置为 [count[0]] 与 [2, count[0]]。 3. **NonZeroWithValueShapeV2**：V2 版本，在 V1 基础上新增 GetInputTypeAndCheck 显式校验 index/count 必须为 INT32，输出语义与 V1 一致。 每个算子按 ops-nn 标准目录结构组织，包含 op_graph（IR 原型）、op_host（infershape）、op_kernel_aicpu（kernel 实现与 REGISTER_CPU_KERNEL 注册）、examples（图模式调用样例）、tests/ut（单元测试）、各级 CMakeLists.txt 与 README.md。 ## 关联的Issue #4163 ## 测试 根据代码变更，测试场景如下： 1. **编译构建测试** - 对三个算子（IndexToAddr / NonZeroWithValueShape / NonZeroWithValueShapeV2）分别执行各级 CMakeLists.txt 配置与编译，验证新增 CMake 配置变更正确，算子可正常构建。 2. **算子加载/注册验证** - 验证 REGISTER_CPU_KERNEL 注册成功，三个算子可被框架正确识别与加载。 3. **InferShape 验证** - 验证 op_host/*_infershape.cpp 输出 shape 推导：IndexToAddr 输出 [block_size[0], 4]；NonZeroWithValueShape[V2] 输出 out_value=[count]、out_index=[2, count]。 4. **单元测试** - 运行 tests/ut/op_kernel_aicpu/test_*.cpp，覆盖 INT64/UINT64（IndexToAddr）、INT32 校验（NonZeroWithValueShapeV2）等数据类型分支与参数校验逻辑。 5. **图模式接口调用验证** - 运行 examples/test_geir_*.cpp，通过算子 IR 构图方式调用三个算子，验证端到端功能与输出正确性。 6. **参数校验测试** - 覆盖 Check() 中的边界场景：base_addr/x 数据类型一致且 shape=[2]、ori_shape/block_size 为 2 元素、block_size[0]*4 <= output.NumElements()、storage mode 仅支持 Matrix、V2 版本 index/count 必须为 DT_INT32、count 至少 1 个元素等。 7. **算子清单登记验证** - 确认 docs/zh/op_list.md 中三个算子条目正确显示，分类为 index，类型为 AI CPU，且 README 链接可达。 ## 文档更新 - 新增 index/index_to_addr/README.md、index/non_zero_with_value_shape/README.md、index/non_zero_with_value_shape_v2/README.md：产品支持情况、功能说明、参数说明、约束说明与调用说明。 - 更新 docs/zh/op_list.md：登记 IndexToAddr、NonZeroWithValueShape、NonZeroWithValueShapeV2 三个算子条目。 ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： ## AI/Agent生成声明 <!-- [x] 表示选中 --> - [ ] AI辅助编写 See merge request: cann/ops-nn!7519 | 1 个月前 | |
Migrate index AICPU shape operators Co-authored-by: jialimin1<jialimin1@huawei.com> # message auto-generated for no-merge-commit merge: !7519 merge migrate-index-to-addr-aicpu into master Migrate index AICPU shape operators Created-by: jialimin1 Commit-by: jialimin1 Merged-by: cann-robot Description: ## 描述 将三个 AICPU shape 类算子从源仓迁移至 ops-nn 仓库，统一算子仓管理，共新增 43 个文件，并同步更新算子清单 docs/zh/op_list.md： 1. **IndexToAddr**：根据块索引、原始矩阵 shape、块 shape 和基地址信息，生成块内每一行对应的地址表。输入 base_addr（shape=[2]）与块索引 x（shape=[2]），输出地址表 shape 为 [block_size[0], 4]，每行 [rank_id, base_addr[0]+bias_addr, base_addr[1]+bias_addr, row_size*dtype_size]，支持 INT64/UINT64。 2. **NonZeroWithValueShape**：根据 NonZeroWithValue 的 count 输入更新 value/index 输出 shape，分别置为 [count[0]] 与 [2, count[0]]。 3. **NonZeroWithValueShapeV2**：V2 版本，在 V1 基础上新增 GetInputTypeAndCheck 显式校验 index/count 必须为 INT32，输出语义与 V1 一致。 每个算子按 ops-nn 标准目录结构组织，包含 op_graph（IR 原型）、op_host（infershape）、op_kernel_aicpu（kernel 实现与 REGISTER_CPU_KERNEL 注册）、examples（图模式调用样例）、tests/ut（单元测试）、各级 CMakeLists.txt 与 README.md。 ## 关联的Issue #4163 ## 测试 根据代码变更，测试场景如下： 1. **编译构建测试** - 对三个算子（IndexToAddr / NonZeroWithValueShape / NonZeroWithValueShapeV2）分别执行各级 CMakeLists.txt 配置与编译，验证新增 CMake 配置变更正确，算子可正常构建。 2. **算子加载/注册验证** - 验证 REGISTER_CPU_KERNEL 注册成功，三个算子可被框架正确识别与加载。 3. **InferShape 验证** - 验证 op_host/*_infershape.cpp 输出 shape 推导：IndexToAddr 输出 [block_size[0], 4]；NonZeroWithValueShape[V2] 输出 out_value=[count]、out_index=[2, count]。 4. **单元测试** - 运行 tests/ut/op_kernel_aicpu/test_*.cpp，覆盖 INT64/UINT64（IndexToAddr）、INT32 校验（NonZeroWithValueShapeV2）等数据类型分支与参数校验逻辑。 5. **图模式接口调用验证** - 运行 examples/test_geir_*.cpp，通过算子 IR 构图方式调用三个算子，验证端到端功能与输出正确性。 6. **参数校验测试** - 覆盖 Check() 中的边界场景：base_addr/x 数据类型一致且 shape=[2]、ori_shape/block_size 为 2 元素、block_size[0]*4 <= output.NumElements()、storage mode 仅支持 Matrix、V2 版本 index/count 必须为 DT_INT32、count 至少 1 个元素等。 7. **算子清单登记验证** - 确认 docs/zh/op_list.md 中三个算子条目正确显示，分类为 index，类型为 AI CPU，且 README 链接可达。 ## 文档更新 - 新增 index/index_to_addr/README.md、index/non_zero_with_value_shape/README.md、index/non_zero_with_value_shape_v2/README.md：产品支持情况、功能说明、参数说明、约束说明与调用说明。 - 更新 docs/zh/op_list.md：登记 IndexToAddr、NonZeroWithValueShape、NonZeroWithValueShapeV2 三个算子条目。 ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： ## AI/Agent生成声明 <!-- [x] 表示选中 --> - [ ] AI辅助编写 See merge request: cann/ops-nn!7519 | 1 个月前 |
NonZeroWithValueShape
产品支持情况
| 产品 | 是否支持 |
|---|---|
| Ascend 950PR/Ascend 950DT | √ |
| Atlas A3 训练系列产品/Atlas A3 推理系列产品 | √ |
| Atlas A2 训练系列产品/Atlas A2 推理系列产品 | √ |
| Atlas 200I/500 A2 推理产品 | × |
| Atlas 推理系列产品 | √ |
| Atlas 训练系列产品 | √ |
功能说明
- 算子功能：根据
NonZeroWithValue的count输入更新value和index输出shape。
参数说明
| 参数名 | 输入/输出 | 描述 | 数据类型 | 数据格式 |
|---|---|---|---|---|
value |
输入 | NonZeroWithValue的value输出。 |
DOUBLE、FLOAT、FLOAT16、INT8、UINT8、INT16、UINT16、INT32、UINT32、INT64、UINT64、BOOL | ND |
index |
输入 | NonZeroWithValue的index输出。 |
INT32 | ND |
count |
输入 | 非零元素数量。 | INT32 | ND |
out_value |
输出 | 更新shape后的value输出，shape为[count[0]]。 |
DOUBLE、FLOAT、FLOAT16、INT8、UINT8、INT16、UINT16、INT32、UINT32、INT64、UINT64、BOOL | ND |
out_index |
输出 | 更新shape后的index输出，shape为[2, count[0]]。 |
INT32 | ND |
约束说明
index和count的数据类型必须为INT32。count输入至少包含1个INT32元素。
调用说明
| 调用方式 | 调用样例 | 说明 |
|---|---|---|
| 图模式调用 | test_geir_non_zero_with_value_shape | 通过算子IR构图方式调用NonZeroWithValueShape算子。 |