已合并
950支持AdaptiveAvgPool3D big kernel模板,解决当kernel Size大于128时,走Simt模板性能较差 #1572
SimonZzz创建于 2月5日
950支持AdaptiveAvgPool3D big kernel模板,解决当kernel Size大于128时,走Simt模板性能较差 #1572
已合并
SimonZzz创建于 2月5日
SimonZzz成员
2月5日

描述

950支持AdaptiveAvgPool3D big kernel模板,解决当kernel Size大于128时,走Simt模板性能较差

关联的Issue

https://gitcode.com/cann/ops-nn/issues/1795

测试

本地测试通过,冒烟通过

文档更新

不涉及

类型标签

likedislike
Pull Request已成功合入, 合并人@CANN-robot
(感谢 SimonZzz 的贡献)
CANN-robotCANN-robot成员
2月5日 将crystalhu,zhou-qilong,chenqi317,liubo75,tangweiwei2,zhangyuxiang0119,wangyongguang,zhang-wu设为审查人
CANN-robot
CANN-robot成员
2月5日 评论:

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 /approve or /lgtm
  • Commenting /approve implies 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. 👍

likedislike
CANN-robot
CANN-robot成员
2月5日 评论:

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

likedislike
CANN-robotCANN-robot成员
2月5日 添加了label:cann-cla/yes
CANN-robot
CANN-robot成员
2月5日 评论:

问题/功能描述

为自适应平均池化3D算子新增对大尺寸池化核(体积DHW >= 128)的支持。原实现在处理大核时可能存在内存访问和计算效率瓶颈,本次修改旨在通过引入分片(Tiling)机制来优化此类场景下的算子性能。

修改方案描述

新增了 AdaptiveAvgPool3dBigKernelTiling 类,继承自基础分片类,专门用于大核场景。核心实现包括:1)通过阈值判断自动启用大核分片模板;2)根据硬件核心数、UB大小等关键因素动态计算分块因子和最大处理元素数;3)精确计算每个输出维度对应的池化核大小;4)将分片参数封装到特定数据结构供内核使用。同时,在基础类中暂时注释了部分索引类型校验代码(标记为TODO),需关注后续处理。

likedislike
此处折叠了177条消息 查看更多
CANN-robot
CANN-robot成员
3月30日 评论:

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.

likedislike
CANN-robotCANN-robot成员
3月30日 关闭了关联的issue
CANN-robotCANN-robot成员
3月30日 合入了pull request
SSimonZzz成员
3月30日 修改了pull request 的描述
SSimonZzz成员
3月30日 修改标题为 “950支持AdaptiveAvgPool3D big kernel模板,解决当kernel Size大于128时,走Simt模板性能较差”,原标题为“add op adaptAvgPool3D Bigkernel”