已合并
AvgPool3D和AvgPool3DGrad两个算子增加infer Datatype #8395
duxinlei创建于 8月7日
AvgPool3D和AvgPool3DGrad两个算子增加infer Datatype #8395
已合并
Pull Request已成功合入, 合并人@CANN-robot
(感谢 duxinlei 的贡献)atomgit-bot
8月7日 评论:
8月7日 评论:
变更摘要
本次变更为 AvgPool3D 和 AvgPool3DGrad 两个算子补充了数据类型推导(InferDataType)能力。AvgPool3D 将输出数据类型设置为与输入 x 一致;AvgPool3DGrad 则将输出数据类型设置为与梯度输入(索引 1)一致。同时为两个算子新增了覆盖 FP16、FP32、BF16 三种数据类型的单元测试。
主要改动
AvgPool3D新增InferDataTypeForAvgPool3D函数:从输入x(索引 0)获取数据类型,直接透传至输出,并在IMPL_OP_INFERSHAPE中通过.InferDataType()注册该推导函数。AvgPool3DGrad新增InferDataTypeForAvgPool3DGrad函数:从梯度输入(索引IDX_1)获取数据类型,设置为输出数据类型,并在IMPL_OP_INFERSHAPE中通过.InferDataType()注册。- 新增单元测试覆盖多精度场景:为
AvgPool3D和AvgPool3DGrad各新增三个测试用例(InferDataTypeFP16、InferDataTypeFP32、InferDataTypeBF16),验证推导结果与预期数据类型一致。


不准确?
atomgit-bot
8月7日 评论:
8月7日 评论:
8月7日 添加了label:cann-cla/yes
CANN-robot
8月7日 评论:
8月7日 评论:
Thanks for your pull-request.
The full list of commands accepted by me can be found at here.
You can get sig-info at here.
You can self-configure the PR merge rules for this repository. For more details, please refer to Here.
For more, you also can visit HICANN.
PR Approval Progress
✅ Congratulations! All modules have met the lgtm and approve requirements.
Module Approval Details
| module | lgtm status | approve status |
|---|---|---|
| pooling | ✅ 黄迪, 刘闯 (2/2) | ✅ 黄迪, 刘闯 (2/1) |
💡 Tip:
- Committer can comment
/approveor/lgtm- Commenting
/approveimplies both code review (lgtm) and intent to merge (approve)
CLA Signature Pass
duxinlei, thanks for your pull request. All authors of the commits have signed the CLA. 👍


此处折叠了146条消息 查看更多
13 天前 添加了label:approved
liuchuangdev
13 天前 评论:
13 天前 评论:
/approve


13 天前 添加了label:lgtm
13 天前 关闭了关联的issue
13 天前 合入了pull request
描述
AvgPool3D和AvgPool3DGrad两个算子增加infer Datatype
关联的Issue
关联issue #5461
测试
新增ut本地编译通过;
出包验证: 蓝区geir通路通过
冒烟通过
文档更新
类型标签
AI/Agent生成声明