已关闭
新增AvgPool1D算子 #4821
wkkk0528创建于  25 天前关闭于  23 天前
wkkk0528
25 天前 创建

描述

AvgPool1D算子支持Ascend950

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

测试

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

文档更新

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

类型标签

likedislike
yuning_chenyuning_chen成员
23 天前 将 wkkk0528 设为负责人
CANN-robotCANN-robot成员
23 天前 关闭了 issue
CANN-robotCANN-robot成员
23 天前 添加了label:resolved