Pull Request已成功合入, 合并人@CANN-robot
(感谢 SimonZzz 的贡献)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
PR Approval Progress
✅ Congratulations! All modules have met the lgtm and approve requirements.
Module Approval Details
| module | lgtm status | approve status |
|---|---|---|
| pooling | ✅ tangweiwei2, chenqi317 (2/2) | ✅ tangweiwei2, chenqi317 (2/1) |
💡 Tip:
- Committer can comment
/approveor/lgtm- Commenting
/approveimplies both code review (lgtm) and intent to merge (approve)
CLA Signature Pass
SimonZzz, thanks for your pull request. All authors of the commits have signed the CLA. 👍


🔵 source code change are detected, tasks labels is removed in this pull request!


问题/功能描述
为自适应平均池化3D算子新增对大尺寸池化核(体积DHW >= 128)的支持。原实现在处理大核时可能存在内存访问和计算效率瓶颈,本次修改旨在通过引入分片(Tiling)机制来优化此类场景下的算子性能。
修改方案描述
新增了 AdaptiveAvgPool3dBigKernelTiling 类,继承自基础分片类,专门用于大核场景。核心实现包括:1)通过阈值判断自动启用大核分片模板;2)根据硬件核心数、UB大小等关键因素动态计算分块因子和最大处理元素数;3)精确计算每个输出维度对应的池化核大小;4)将分片参数封装到特定数据结构供内核使用。同时,在基础类中暂时注释了部分索引类型校验代码(标记为TODO),需关注后续处理。


Review Guide
This pull-request passes review.
Committers who wrote a comment of /approve are: tangweiwei2, chenqi317.
Reviewers who wrote a comment of /lgtm are: tangweiwei2, chenqi317.


描述
950支持AdaptiveAvgPool3D big kernel模板,解决当kernel Size大于128时,走Simt模板性能较差
关联的Issue
https://gitcode.com/cann/ops-nn/issues/1795
测试
本地测试通过,冒烟通过
文档更新
不涉及
类型标签