| 文件 | 最后提交记录 | 最后更新时间 |
|---|---|---|
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 个月前 |
IndexToAddr
产品支持情况
| 产品 | 是否支持 |
|---|---|
| Ascend 950PR/Ascend 950DT | √ |
| Atlas A3 训练系列产品/Atlas A3 推理系列产品 | √ |
| Atlas A2 训练系列产品/Atlas A2 推理系列产品 | √ |
| Atlas 200I/500 A2 推理产品 | × |
| Atlas 推理系列产品 | √ |
| Atlas 训练系列产品 | √ |
功能说明
- 算子功能：根据块索引、原始矩阵shape、块shape和基地址信息，生成块内每一行对应的地址表。
参数说明
| 参数名 | 输入/输出/属性 | 描述 | 数据类型 | 数据格式 |
|---|---|---|---|---|
base_addr |
输入 | 基地址张量，shape为[2]。 |
INT64、UINT64 | ND |
x |
输入 | 块索引张量，shape为[2]，第0个元素为块行索引，第1个元素为块列索引。 |
INT64、UINT64 | ND |
ori_shape |
属性 | 原始矩阵shape，长度为2。 | ListInt | - |
block_size |
属性 | 块shape，长度为2。 | ListInt | - |
ori_storage_mode |
属性 | 原始矩阵存储模式，默认Matrix。当前仅支持Matrix。 |
STRING | - |
block_storage_mode |
属性 | 块存储模式，默认Matrix。当前仅支持Matrix。 |
STRING | - |
rank_id |
属性 | rank id，默认0。 | INT | - |
dtype |
属性 | 基础数据类型，默认DT_FLOAT。 | TYPE | - |
addrs_table |
输出 | 地址表张量，shape为[block_size[0], 4]。 |
INT64、UINT64 | ND |
约束说明
base_addr和x的数据类型必须一致。base_addr和x必须是一维且元素个数为2的张量。ori_shape和block_size必须是长度为2的ListInt。ori_storage_mode和block_storage_mode当前仅支持Matrix。x[0] * block_size[0]必须小于ori_shape[0]，x[1] * block_size[1]必须小于ori_shape[1]。- 输出张量元素个数必须不小于
block_size[0] * 4。
调用说明
| 调用方式 | 调用样例 | 说明 |
|---|---|---|
| 图模式调用 | test_geir_index_to_addr | 通过算子IR构图方式调用IndexToAddr算子。 |