| 文件 | 最后提交记录 | 最后更新时间 |
|---|---|---|
docs目录基本概念md文件名改为英文 Co-authored-by: chenjiao<chenjiao31@huawei.com> # message auto-generated for no-merge-commit merge: !4182 merge master into master docs目录基本概念md文件名改为英文 Created-by: gitcode-chenjiao Commit-by: chenjiao Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动，包括改动的原因和所采取的方法。--> docs目录基本概念md文件名改为英文： 避免link中的中文字符引发的跳转异常，例如两段式接口.md变成%E4%B8%A4%E6%AE%B5%E5%BC%8F%E6%8E%A5%E5%8F%A3.md，不易于维护，可能导致其他平台跳转有问题。 ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。--> <!-- 如果这个PR是为了解决特定的问题单，请在这里描述问题单单号。--> [#2283](https://gitcode.com/cann/ops-math/issues/2283) ## 测试 <!--描述进行了哪些测试来验证你的改动。包括但不限于二级冒烟、算子泛化等。--> ok ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> docs/zh/context所有md和对应的link ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [x ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!4182 | 22 天前 | |
add aicpu:topk&calc_buckets_limit_and_offset&compare_and_bit_pack Co-authored-by: sujunwei3<sujunwei3@huawei.com> # message auto-generated for no-merge-commit merge: !3724 merge dev into master add aicpu:topk&calc_buckets_limit_and_offset&compare_and_bit_pack Created-by: sujunwei3 Commit-by: sujunwei3 Merged-by: cann-robot Description: ## 描述 新增 TopK、CalcBucketsLimitAndOffset、CompareAndBitPack 三个 AICPU 算子的完整实现，共涉及 28 个文件变更。 ### 1. TopK（math/topk/） 从已有 aclnn 接口扩展为完整 AICPU 算子实现： - 新增 op_graph/topk_proto.h：算子协议定义 - 新增 op_host/topk_infershape.cpp：host 侧 shape 推导 - 新增 op_kernel_aicpu/topk_aicpu.cpp、topk_aicpu.h、topk_aicpu_def.cpp：AICPU kernel 实现及注册 - 新增 op_kernel_aicpu/CMakeLists.txt、op_graph/CMakeLists.txt：构建配置 - 新增 examples/test_geir_topk.cpp：使用示例 - 新增 tests/ut/op_kernel_aicpu/test_topk.cpp：单元测试 - 修改 CMakeLists.txt：添加 UT 测试配置 - 修改 README.md：补充完整算子文档（产品支持情况、接口说明等） ### 2. CalcBucketsLimitAndOffset（conversion/calc_buckets_limit_and_offset/） 全新 AICPU 算子，用于计算 buckets 的 limit 和 offset 值： - 新增 CMakeLists.txt、op_graph/calc_buckets_limit_and_offset_proto.h - 新增 op_kernel_aicpu/calc_buckets_limit_and_offset_aicpu.cpp、.h、_def.cpp - 新增 examples/test_geir_calc_buckets_limit_and_offset.cpp - 新增 tests/ut/op_kernel_aicpu/test_calc_buckets_limit_and_offset.cpp - 新增 README.md：算子文档 ### 3. CompareAndBitPack（math/compare_and_bit_pack/） 全新 AICPU 算子，将输入与阈值进行比较并将比较结果的比特位打包为 uint8 输出： - 新增 CMakeLists.txt、op_graph/compare_and_bit_pack_proto.h - 新增 op_kernel_aicpu/compare_and_bit_pack_aicpu.cpp、.h、_def.cpp - 新增 examples/test_geir_compare_and_bit_pack.cpp - 新增 tests/ut/op_kernel_aicpu/test_compare_and_bit_pack.cpp - 新增 README.md：算子文档 ### 4. 文档更新 - 修改 docs/zh/op_list.md：在算子列表中添加 compare_and_bit_pack 等算子条目 ## 关联的Issue #2187 ## 测试 根据代码变更，测试场景如下： 1. **算子构建测试** - 执行 cmake 配置和编译，验证三个算子（TopK、CalcBucketsLimitAndOffset、CompareAndBitPack）的 CMakeLists.txt 配置正确，能正常编译构建 2. **单元测试（UT）** - 运行 tests/ut/op_kernel_aicpu/test_topk.cpp，验证 TopK 算子在不同数据类型（float16、float32、int32）和 shape 下的计算正确性 - 运行 tests/ut/op_kernel_aicpu/test_calc_buckets_limit_and_offset.cpp，验证 CalcBucketsLimitAndOffset 算子的功能正确性 - 运行 tests/ut/op_kernel_aicpu/test_compare_and_bit_pack.cpp，验证 CompareAndBitPack 算子在不同数据类型下的比较和比特打包功能正确性 3. **Example 验证** - 编译并运行 examples/test_geir_topk.cpp，验证 TopK 算子端到端调用流程 - 编译并运行 examples/test_geir_calc_buckets_limit_and_offset.cpp，验证 CalcBucketsLimitAndOffset 算子端到端调用流程 - 编译并运行 examples/test_geir_compare_and_bit_pack.cpp，验证 CompareAndBitPack 算子端到端调用流程 4. **算子加载测试** - 验证三个算子的 AICPU kernel 注册成功，接口可正常调用 ## 文档更新 - 新增 math/topk/README.md：TopK 算子完整文档（产品支持情况、接口说明、约束限制等） - 新增 conversion/calc_buckets_limit_and_offset/README.md：CalcBucketsLimitAndOffset 算子文档 - 新增 math/compare_and_bit_pack/README.md：CompareAndBitPack 算子文档 - 修改 docs/zh/op_list.md：算子列表新增条目 ## 类型标签 - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [x] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!3724 | 1 个月前 | |
add aicpu:topk&calc_buckets_limit_and_offset&compare_and_bit_pack Co-authored-by: sujunwei3<sujunwei3@huawei.com> # message auto-generated for no-merge-commit merge: !3724 merge dev into master add aicpu:topk&calc_buckets_limit_and_offset&compare_and_bit_pack Created-by: sujunwei3 Commit-by: sujunwei3 Merged-by: cann-robot Description: ## 描述 新增 TopK、CalcBucketsLimitAndOffset、CompareAndBitPack 三个 AICPU 算子的完整实现，共涉及 28 个文件变更。 ### 1. TopK（math/topk/） 从已有 aclnn 接口扩展为完整 AICPU 算子实现： - 新增 op_graph/topk_proto.h：算子协议定义 - 新增 op_host/topk_infershape.cpp：host 侧 shape 推导 - 新增 op_kernel_aicpu/topk_aicpu.cpp、topk_aicpu.h、topk_aicpu_def.cpp：AICPU kernel 实现及注册 - 新增 op_kernel_aicpu/CMakeLists.txt、op_graph/CMakeLists.txt：构建配置 - 新增 examples/test_geir_topk.cpp：使用示例 - 新增 tests/ut/op_kernel_aicpu/test_topk.cpp：单元测试 - 修改 CMakeLists.txt：添加 UT 测试配置 - 修改 README.md：补充完整算子文档（产品支持情况、接口说明等） ### 2. CalcBucketsLimitAndOffset（conversion/calc_buckets_limit_and_offset/） 全新 AICPU 算子，用于计算 buckets 的 limit 和 offset 值： - 新增 CMakeLists.txt、op_graph/calc_buckets_limit_and_offset_proto.h - 新增 op_kernel_aicpu/calc_buckets_limit_and_offset_aicpu.cpp、.h、_def.cpp - 新增 examples/test_geir_calc_buckets_limit_and_offset.cpp - 新增 tests/ut/op_kernel_aicpu/test_calc_buckets_limit_and_offset.cpp - 新增 README.md：算子文档 ### 3. CompareAndBitPack（math/compare_and_bit_pack/） 全新 AICPU 算子，将输入与阈值进行比较并将比较结果的比特位打包为 uint8 输出： - 新增 CMakeLists.txt、op_graph/compare_and_bit_pack_proto.h - 新增 op_kernel_aicpu/compare_and_bit_pack_aicpu.cpp、.h、_def.cpp - 新增 examples/test_geir_compare_and_bit_pack.cpp - 新增 tests/ut/op_kernel_aicpu/test_compare_and_bit_pack.cpp - 新增 README.md：算子文档 ### 4. 文档更新 - 修改 docs/zh/op_list.md：在算子列表中添加 compare_and_bit_pack 等算子条目 ## 关联的Issue #2187 ## 测试 根据代码变更，测试场景如下： 1. **算子构建测试** - 执行 cmake 配置和编译，验证三个算子（TopK、CalcBucketsLimitAndOffset、CompareAndBitPack）的 CMakeLists.txt 配置正确，能正常编译构建 2. **单元测试（UT）** - 运行 tests/ut/op_kernel_aicpu/test_topk.cpp，验证 TopK 算子在不同数据类型（float16、float32、int32）和 shape 下的计算正确性 - 运行 tests/ut/op_kernel_aicpu/test_calc_buckets_limit_and_offset.cpp，验证 CalcBucketsLimitAndOffset 算子的功能正确性 - 运行 tests/ut/op_kernel_aicpu/test_compare_and_bit_pack.cpp，验证 CompareAndBitPack 算子在不同数据类型下的比较和比特打包功能正确性 3. **Example 验证** - 编译并运行 examples/test_geir_topk.cpp，验证 TopK 算子端到端调用流程 - 编译并运行 examples/test_geir_calc_buckets_limit_and_offset.cpp，验证 CalcBucketsLimitAndOffset 算子端到端调用流程 - 编译并运行 examples/test_geir_compare_and_bit_pack.cpp，验证 CompareAndBitPack 算子端到端调用流程 4. **算子加载测试** - 验证三个算子的 AICPU kernel 注册成功，接口可正常调用 ## 文档更新 - 新增 math/topk/README.md：TopK 算子完整文档（产品支持情况、接口说明、约束限制等） - 新增 conversion/calc_buckets_limit_and_offset/README.md：CalcBucketsLimitAndOffset 算子文档 - 新增 math/compare_and_bit_pack/README.md：CompareAndBitPack 算子文档 - 修改 docs/zh/op_list.md：算子列表新增条目 ## 类型标签 - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [x] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!3724 | 1 个月前 | |
add aicpu:topk&calc_buckets_limit_and_offset&compare_and_bit_pack Co-authored-by: sujunwei3<sujunwei3@huawei.com> # message auto-generated for no-merge-commit merge: !3724 merge dev into master add aicpu:topk&calc_buckets_limit_and_offset&compare_and_bit_pack Created-by: sujunwei3 Commit-by: sujunwei3 Merged-by: cann-robot Description: ## 描述 新增 TopK、CalcBucketsLimitAndOffset、CompareAndBitPack 三个 AICPU 算子的完整实现，共涉及 28 个文件变更。 ### 1. TopK（math/topk/） 从已有 aclnn 接口扩展为完整 AICPU 算子实现： - 新增 op_graph/topk_proto.h：算子协议定义 - 新增 op_host/topk_infershape.cpp：host 侧 shape 推导 - 新增 op_kernel_aicpu/topk_aicpu.cpp、topk_aicpu.h、topk_aicpu_def.cpp：AICPU kernel 实现及注册 - 新增 op_kernel_aicpu/CMakeLists.txt、op_graph/CMakeLists.txt：构建配置 - 新增 examples/test_geir_topk.cpp：使用示例 - 新增 tests/ut/op_kernel_aicpu/test_topk.cpp：单元测试 - 修改 CMakeLists.txt：添加 UT 测试配置 - 修改 README.md：补充完整算子文档（产品支持情况、接口说明等） ### 2. CalcBucketsLimitAndOffset（conversion/calc_buckets_limit_and_offset/） 全新 AICPU 算子，用于计算 buckets 的 limit 和 offset 值： - 新增 CMakeLists.txt、op_graph/calc_buckets_limit_and_offset_proto.h - 新增 op_kernel_aicpu/calc_buckets_limit_and_offset_aicpu.cpp、.h、_def.cpp - 新增 examples/test_geir_calc_buckets_limit_and_offset.cpp - 新增 tests/ut/op_kernel_aicpu/test_calc_buckets_limit_and_offset.cpp - 新增 README.md：算子文档 ### 3. CompareAndBitPack（math/compare_and_bit_pack/） 全新 AICPU 算子，将输入与阈值进行比较并将比较结果的比特位打包为 uint8 输出： - 新增 CMakeLists.txt、op_graph/compare_and_bit_pack_proto.h - 新增 op_kernel_aicpu/compare_and_bit_pack_aicpu.cpp、.h、_def.cpp - 新增 examples/test_geir_compare_and_bit_pack.cpp - 新增 tests/ut/op_kernel_aicpu/test_compare_and_bit_pack.cpp - 新增 README.md：算子文档 ### 4. 文档更新 - 修改 docs/zh/op_list.md：在算子列表中添加 compare_and_bit_pack 等算子条目 ## 关联的Issue #2187 ## 测试 根据代码变更，测试场景如下： 1. **算子构建测试** - 执行 cmake 配置和编译，验证三个算子（TopK、CalcBucketsLimitAndOffset、CompareAndBitPack）的 CMakeLists.txt 配置正确，能正常编译构建 2. **单元测试（UT）** - 运行 tests/ut/op_kernel_aicpu/test_topk.cpp，验证 TopK 算子在不同数据类型（float16、float32、int32）和 shape 下的计算正确性 - 运行 tests/ut/op_kernel_aicpu/test_calc_buckets_limit_and_offset.cpp，验证 CalcBucketsLimitAndOffset 算子的功能正确性 - 运行 tests/ut/op_kernel_aicpu/test_compare_and_bit_pack.cpp，验证 CompareAndBitPack 算子在不同数据类型下的比较和比特打包功能正确性 3. **Example 验证** - 编译并运行 examples/test_geir_topk.cpp，验证 TopK 算子端到端调用流程 - 编译并运行 examples/test_geir_calc_buckets_limit_and_offset.cpp，验证 CalcBucketsLimitAndOffset 算子端到端调用流程 - 编译并运行 examples/test_geir_compare_and_bit_pack.cpp，验证 CompareAndBitPack 算子端到端调用流程 4. **算子加载测试** - 验证三个算子的 AICPU kernel 注册成功，接口可正常调用 ## 文档更新 - 新增 math/topk/README.md：TopK 算子完整文档（产品支持情况、接口说明、约束限制等） - 新增 conversion/calc_buckets_limit_and_offset/README.md：CalcBucketsLimitAndOffset 算子文档 - 新增 math/compare_and_bit_pack/README.md：CompareAndBitPack 算子文档 - 修改 docs/zh/op_list.md：算子列表新增条目 ## 类型标签 - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [x] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!3724 | 1 个月前 | |
fix compile Co-authored-by: sujunwei3<sujunwei3@huawei.com> # message auto-generated for no-merge-commit merge: !3993 merge dev into master fix compile Created-by: sujunwei3 Commit-by: sujunwei3 Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动，包括改动的原因和所采取的方法。--> 修改编包失败问题 ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。--> <!-- 如果这个PR是为了解决特定的问题单，请在这里描述问题单单号。--> ## 测试 <!--描述进行了哪些测试来验证你的改动。包括但不限于二级冒烟、算子泛化等。--> ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [x] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!3993 | 1 个月前 | |
fix: 修复topk/calc_buckets_limit_and_offset/compare_and_bit_pack算子UT问题 Co-authored-by: sujunwei3<sujunwei3@huawei.com> # message auto-generated for no-merge-commit merge: !4173 merge fix/ut-issues into master fix: 修复topk/calc_buckets_limit_and_offset/compare_and_bit_pack算子UT问题 Created-by: sujunwei3 Commit-by: sujunwei3 Merged-by: cann-robot Description: ## 描述 修复 topk、calc_buckets_limit_and_offset、compare_and_bit_pack 三个算子的 UT 问题及内核参数校验缺失。 **calc_buckets_limit_and_offset 算子：** - 内核代码新增 total_limit >= 0 参数校验和 bucket_list[i] >= 0 输入范围检查 - 修复 UT 中 input2_num 赋值 typo（input1->NumElements() → input2->NumElements()） **compare_and_bit_pack 算子：** - COMPLEX64 用例重命名为 INPUT_COMPLEX64_UNSUPPORTED，移除无效输出验证 - BOOL 用例修正 input1 值（true → false）及期望结果 - DIM_ERROR 用例修正 shape 和输入数据指针 **topk 算子：** - 重写 SMALLEST_LAST_DIM 测试：修正输出维度、使用随机值填充、按行排序取 top-2、添加完整结果校验 ## 关联的Issue #2285 ## 测试 1. **单元测试验证** - 执行 calc_buckets_limit_and_offset 算子 UT，验证参数校验和范围检查生效 - 执行 compare_and_bit_pack 算子 UT，验证 COMPLEX64/BOOL/DIM_ERROR 用例通过 - 执行 topk 算子 UT，验证 SMALLEST_LAST_DIM 用例排序结果正确 2. **回归测试** - 确保修改不影响其他已有 UT 用例的通过 ## 文档更新 无文档更新 ## 类型标签 - [x] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!4173 | 22 天前 | |
add aicpu:topk&calc_buckets_limit_and_offset&compare_and_bit_pack Co-authored-by: sujunwei3<sujunwei3@huawei.com> # message auto-generated for no-merge-commit merge: !3724 merge dev into master add aicpu:topk&calc_buckets_limit_and_offset&compare_and_bit_pack Created-by: sujunwei3 Commit-by: sujunwei3 Merged-by: cann-robot Description: ## 描述 新增 TopK、CalcBucketsLimitAndOffset、CompareAndBitPack 三个 AICPU 算子的完整实现，共涉及 28 个文件变更。 ### 1. TopK（math/topk/） 从已有 aclnn 接口扩展为完整 AICPU 算子实现： - 新增 op_graph/topk_proto.h：算子协议定义 - 新增 op_host/topk_infershape.cpp：host 侧 shape 推导 - 新增 op_kernel_aicpu/topk_aicpu.cpp、topk_aicpu.h、topk_aicpu_def.cpp：AICPU kernel 实现及注册 - 新增 op_kernel_aicpu/CMakeLists.txt、op_graph/CMakeLists.txt：构建配置 - 新增 examples/test_geir_topk.cpp：使用示例 - 新增 tests/ut/op_kernel_aicpu/test_topk.cpp：单元测试 - 修改 CMakeLists.txt：添加 UT 测试配置 - 修改 README.md：补充完整算子文档（产品支持情况、接口说明等） ### 2. CalcBucketsLimitAndOffset（conversion/calc_buckets_limit_and_offset/） 全新 AICPU 算子，用于计算 buckets 的 limit 和 offset 值： - 新增 CMakeLists.txt、op_graph/calc_buckets_limit_and_offset_proto.h - 新增 op_kernel_aicpu/calc_buckets_limit_and_offset_aicpu.cpp、.h、_def.cpp - 新增 examples/test_geir_calc_buckets_limit_and_offset.cpp - 新增 tests/ut/op_kernel_aicpu/test_calc_buckets_limit_and_offset.cpp - 新增 README.md：算子文档 ### 3. CompareAndBitPack（math/compare_and_bit_pack/） 全新 AICPU 算子，将输入与阈值进行比较并将比较结果的比特位打包为 uint8 输出： - 新增 CMakeLists.txt、op_graph/compare_and_bit_pack_proto.h - 新增 op_kernel_aicpu/compare_and_bit_pack_aicpu.cpp、.h、_def.cpp - 新增 examples/test_geir_compare_and_bit_pack.cpp - 新增 tests/ut/op_kernel_aicpu/test_compare_and_bit_pack.cpp - 新增 README.md：算子文档 ### 4. 文档更新 - 修改 docs/zh/op_list.md：在算子列表中添加 compare_and_bit_pack 等算子条目 ## 关联的Issue #2187 ## 测试 根据代码变更，测试场景如下： 1. **算子构建测试** - 执行 cmake 配置和编译，验证三个算子（TopK、CalcBucketsLimitAndOffset、CompareAndBitPack）的 CMakeLists.txt 配置正确，能正常编译构建 2. **单元测试（UT）** - 运行 tests/ut/op_kernel_aicpu/test_topk.cpp，验证 TopK 算子在不同数据类型（float16、float32、int32）和 shape 下的计算正确性 - 运行 tests/ut/op_kernel_aicpu/test_calc_buckets_limit_and_offset.cpp，验证 CalcBucketsLimitAndOffset 算子的功能正确性 - 运行 tests/ut/op_kernel_aicpu/test_compare_and_bit_pack.cpp，验证 CompareAndBitPack 算子在不同数据类型下的比较和比特打包功能正确性 3. **Example 验证** - 编译并运行 examples/test_geir_topk.cpp，验证 TopK 算子端到端调用流程 - 编译并运行 examples/test_geir_calc_buckets_limit_and_offset.cpp，验证 CalcBucketsLimitAndOffset 算子端到端调用流程 - 编译并运行 examples/test_geir_compare_and_bit_pack.cpp，验证 CompareAndBitPack 算子端到端调用流程 4. **算子加载测试** - 验证三个算子的 AICPU kernel 注册成功，接口可正常调用 ## 文档更新 - 新增 math/topk/README.md：TopK 算子完整文档（产品支持情况、接口说明、约束限制等） - 新增 conversion/calc_buckets_limit_and_offset/README.md：CalcBucketsLimitAndOffset 算子文档 - 新增 math/compare_and_bit_pack/README.md：CompareAndBitPack 算子文档 - 修改 docs/zh/op_list.md：算子列表新增条目 ## 类型标签 - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [x] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!3724 | 1 个月前 | |
add aicpu:topk&calc_buckets_limit_and_offset&compare_and_bit_pack Co-authored-by: sujunwei3<sujunwei3@huawei.com> # message auto-generated for no-merge-commit merge: !3724 merge dev into master add aicpu:topk&calc_buckets_limit_and_offset&compare_and_bit_pack Created-by: sujunwei3 Commit-by: sujunwei3 Merged-by: cann-robot Description: ## 描述 新增 TopK、CalcBucketsLimitAndOffset、CompareAndBitPack 三个 AICPU 算子的完整实现，共涉及 28 个文件变更。 ### 1. TopK（math/topk/） 从已有 aclnn 接口扩展为完整 AICPU 算子实现： - 新增 op_graph/topk_proto.h：算子协议定义 - 新增 op_host/topk_infershape.cpp：host 侧 shape 推导 - 新增 op_kernel_aicpu/topk_aicpu.cpp、topk_aicpu.h、topk_aicpu_def.cpp：AICPU kernel 实现及注册 - 新增 op_kernel_aicpu/CMakeLists.txt、op_graph/CMakeLists.txt：构建配置 - 新增 examples/test_geir_topk.cpp：使用示例 - 新增 tests/ut/op_kernel_aicpu/test_topk.cpp：单元测试 - 修改 CMakeLists.txt：添加 UT 测试配置 - 修改 README.md：补充完整算子文档（产品支持情况、接口说明等） ### 2. CalcBucketsLimitAndOffset（conversion/calc_buckets_limit_and_offset/） 全新 AICPU 算子，用于计算 buckets 的 limit 和 offset 值： - 新增 CMakeLists.txt、op_graph/calc_buckets_limit_and_offset_proto.h - 新增 op_kernel_aicpu/calc_buckets_limit_and_offset_aicpu.cpp、.h、_def.cpp - 新增 examples/test_geir_calc_buckets_limit_and_offset.cpp - 新增 tests/ut/op_kernel_aicpu/test_calc_buckets_limit_and_offset.cpp - 新增 README.md：算子文档 ### 3. CompareAndBitPack（math/compare_and_bit_pack/） 全新 AICPU 算子，将输入与阈值进行比较并将比较结果的比特位打包为 uint8 输出： - 新增 CMakeLists.txt、op_graph/compare_and_bit_pack_proto.h - 新增 op_kernel_aicpu/compare_and_bit_pack_aicpu.cpp、.h、_def.cpp - 新增 examples/test_geir_compare_and_bit_pack.cpp - 新增 tests/ut/op_kernel_aicpu/test_compare_and_bit_pack.cpp - 新增 README.md：算子文档 ### 4. 文档更新 - 修改 docs/zh/op_list.md：在算子列表中添加 compare_and_bit_pack 等算子条目 ## 关联的Issue #2187 ## 测试 根据代码变更，测试场景如下： 1. **算子构建测试** - 执行 cmake 配置和编译，验证三个算子（TopK、CalcBucketsLimitAndOffset、CompareAndBitPack）的 CMakeLists.txt 配置正确，能正常编译构建 2. **单元测试（UT）** - 运行 tests/ut/op_kernel_aicpu/test_topk.cpp，验证 TopK 算子在不同数据类型（float16、float32、int32）和 shape 下的计算正确性 - 运行 tests/ut/op_kernel_aicpu/test_calc_buckets_limit_and_offset.cpp，验证 CalcBucketsLimitAndOffset 算子的功能正确性 - 运行 tests/ut/op_kernel_aicpu/test_compare_and_bit_pack.cpp，验证 CompareAndBitPack 算子在不同数据类型下的比较和比特打包功能正确性 3. **Example 验证** - 编译并运行 examples/test_geir_topk.cpp，验证 TopK 算子端到端调用流程 - 编译并运行 examples/test_geir_calc_buckets_limit_and_offset.cpp，验证 CalcBucketsLimitAndOffset 算子端到端调用流程 - 编译并运行 examples/test_geir_compare_and_bit_pack.cpp，验证 CompareAndBitPack 算子端到端调用流程 4. **算子加载测试** - 验证三个算子的 AICPU kernel 注册成功，接口可正常调用 ## 文档更新 - 新增 math/topk/README.md：TopK 算子完整文档（产品支持情况、接口说明、约束限制等） - 新增 conversion/calc_buckets_limit_and_offset/README.md：CalcBucketsLimitAndOffset 算子文档 - 新增 math/compare_and_bit_pack/README.md：CompareAndBitPack 算子文档 - 修改 docs/zh/op_list.md：算子列表新增条目 ## 类型标签 - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [x] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!3724 | 1 个月前 |
TopK
产品支持情况
| 产品 | 是否支持 |
|---|---|
| Ascend 950PR/Ascend 950DT | √ |
| Atlas A3 训练系列产品/Atlas A3 推理系列产品 | √ |
| Atlas A2 训练系列产品/Atlas A2 推理系列产品 | √ |
| Atlas 200I/500 A2 推理产品 | √ |
| Atlas 推理系列产品 | √ |
| Atlas 训练系列产品 | √ |
功能说明
- 算子功能：沿指定维度找出输入张量中最大或最小的k个元素及其索引。
参数说明
| 参数名 | 输入/输出/属性 | 描述 | 数据类型 | 数据格式 |
|---|---|---|---|---|
| x | 输入 | 输入张量。 | FLOAT16、FLOAT、DOUBLE、INT8、INT16、INT32、INT64、UINT8、UINT16、UINT32、UINT64、BFLOAT16 | ND |
| k | 输入 | 要取出的元素个数。 | INT32 | ND |
| sorted | 属性 | 是否对输出结果排序，默认为true。 | Bool | - |
| largest | 属性 | 是否取最大值，默认为true。若为false则取最小值。 | Bool | - |
| dim | 属性 | 指定沿哪个维度进行操作，默认为-1（最后一维）。 | Int | - |
| values | 输出 | 输出的top-k个元素值。 | FLOAT16、FLOAT、DOUBLE、INT8、INT16、INT32、INT64、UINT8、UINT16、UINT32、UINT64、BFLOAT16 | ND |
| indices | 输出 | 输出的top-k个元素在输入张量中的索引。 | INT32 | ND |
约束说明
- dim必须在[-x维度数, x维度数)范围内
- k必须大于等于0且小于等于x在dim维度上的大小
- BFLOAT16类型仅在特定产品上支持
调用说明
| 调用方式 | 样例代码 | 说明 |
|---|---|---|
| 图模式接口 | test_geir_topk | 通过算子IR接口方式调用TopK算子。 |