| 文件 | 最后提交记录 | 最后更新时间 |
|---|---|---|
feat: add bucketize runtime2 infershape Co-authored-by: ayuemoon<15802951036@163.com> # message auto-generated for no-merge-commit merge: !9798 merge bucketize-rt2-infershape into master feat: add bucketize runtime2 infershape Created-by: ayuemoon Commit-by: ayuemoon Merged-by: cann-robot Description: ## 描述 Bucketize 算子缺少 Runtime 2（RT2） InferShape 注册，导致图编译阶段无法正确完成输出 shape 和 dtype 推导。本 PR 为 Bucketize 补充 RT2 infershape 实现。 实现内容： - 使用 Ops::Base::InferShape4Elewise，保持输出 shape 与输入 shape 一致，并兼容动态维度。 - 根据 dtype 属性设置输出数据类型，支持 DT_INT32 和 DT_INT64。 - 接入 op_host 源码自动收集和 infershape UT 构建配置。 ## 关联的Issue 无。 ## 测试 在 Ascend950 环境执行 bash build.sh -u --ophost --ops=bucketize --soc=ascend950 -j8，并使用 nn_op_host_ut --gtest_filter=BucketizeInfershapeTest.* 运行 Bucketize InferShape UT，覆盖普通 Tensor、标量、空 Tensor、Unknown Rank、Unknown Dim 及 INT32/INT64 输出类型等场景，全部通过。 ## 文档更新 无。 ## 类型标签 - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： ## AI/Agent生成声明 - [x] AI辅助编写 See merge request: cann/ops-nn!9798 | 1 个月前 | |
feat: migrate bucketize and avg_pool1d_avg_matrix Co-authored-by: zhaowenrui666<zhaowenrui7@huawei.com> # message auto-generated for no-merge-commit merge: !6877 merge migrate-bucketize-avgpool1d into master feat: migrate bucketize and avg_pool1d_avg_matrix Created-by: zhaowenrui666 Commit-by: zhaowenrui666 Merged-by: cann-robot Description: ## Summary 迁移两个算子到 ops-nn 正确分类目录： - Bucketize -> index/bucketize - AvgPool1DAvgMatrix -> pooling/avg_pool1d_avg_matrix 本次按标准 AICPU 迁仓补齐： - AICPU kernel 与 *_aicpu_def.cpp - op_graph / op_host - UT - README - GE IR example ## Validation - bash ./build.sh --run_example bucketize graph --soc=ascend910_93 --noexec - example 运行成功，输出 bucketize 结果并打印 success - bash ./build.sh --run_example avg_pool1d_avg_matrix graph --soc=ascend910_93 --noexec - example 运行成功，输出 avg matrix 结果并打印 success - git diff --cached --check passed before commit See merge request: cann/ops-nn!6877 | 3 个月前 |
| 文件 | 最后提交记录 | 最后更新时间 |
|---|---|---|
| 1 个月前 | ||
| 3 个月前 |