已合并
[CANNBot]新增AvgPool1D算子 #8290
wkkk0528创建于 8月5日
[CANNBot]新增AvgPool1D算子 #8290
已合并
wkkk0528创建于 8月5日
wkkk0528
8月5日

描述

AvgPool1D算子支持Ascend950

AvgPool1D是一维平均池化算子,采用AscendC SIMT实现,支持Grid-Stride多核并行+单线程内顺序确定性累加,不使用原子操作。支持不对称padding、ceil_mode输出长度修正和count_include_pad分母选择。

关联的issue

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

测试

  • 编译验证通过(ascend950 --pkg)
  • TTK Kernel通路测试通过(315用例100%通过)
  • 三方比对测试通过(NPU vs CPU vs GPU)
  • 确定性验证通过(单线程内顺序累加,无atomicAdd)

文档更新

  • 新增README.md
  • 新增op_graph原型定义
  • 更新op_list.md

类型标签

AI/Agent 生成声明

代码来源:

人工审查:已完成
测试验证:已通过
合规检查:已完成

likedislike
Pull Request已成功合入, 合并人@CANN-robot
(感谢 wkkk0528 的贡献)
Wwkkk0528
8月5日 创建了 pull request,commit 69c59a93
atomgit-bot
atomgit-bot
8月5日 评论:

🤖 正在生成合并请求摘要,请稍候…

likedislike
不准确?
atomgit-bot
atomgit-bot
8月5日 评论:

🤖 AI 代码检视正在进行中,请稍候…

likedislike
不准确?
CANN-robotCANN-robot成员
8月5日 添加了label:cann-cla/yes
CANN-robot
CANN-robot成员
8月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.
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
*/*/README.md ✅ 陈娇, 唐玮玮 (2/2) ✅ 陈娇 (1/1)
*/*/op_graph/*_proto.h ✅ 汤磊, 唐玮玮 (2/2) ✅ 汤磊 (1/1)
*/*/op_host/*_def.cpp ✅ 汤磊, 唐玮玮 (2/2) ✅ 汤磊 (1/1)
docs ✅ 陈娇, 唐玮玮 (2/2) ✅ 陈娇 (1/1)
pooling ✅ 刘闯, 黄迪, 唐玮玮 (3/2) ✅ 刘闯, 黄迪 (2/1)

💡 Tip:

  • Committer can comment /approve or /lgtm
  • Commenting /approve implies both code review (lgtm) and intent to merge (approve)

CLA Signature Pass

wkkk0528, thanks for your pull request. All authors of the commits have signed the CLA. 👍

likedislike
此处折叠了292条消息 查看更多
tangweiwei2成员
8月17日 评论:

/lgtm
/approve

likedislike
CANN-robotCANN-robot成员
8月17日 添加了label:lgtm
CANN-robotCANN-robot成员
8月17日 关闭了关联的issue
CANN-robotCANN-robot成员
8月17日 合入了pull request
CANN-robot
CANN-robot成员
8月17日 评论:

Pull Request 已合并或已关闭。

If you want to solve this problem, you can click here to do it in the FAQs.

likedislike