已开启
[Requirement|需求建议]: 新增AvgPoolUpdate算子支持Ascend950 #5152
吴成文创建于 21 天前
21 天前 添加了label:requirement
21 天前 关联了pull request:[CANNBot]AvgPoolUpdate算子支持Ascend950
21 天前 将 wuchengwen_oh 设为负责人
21 天前 关联了pull request:[CANNBot]AvgPoolUpdate算子支持Ascend950
21 天前 关联了pull request:[CANNBot]新增NPUClearFloatStatus与AvgPoolUpdate算子支持Ascend950
Backgroud(背景信息)
AvgPool算子在exclusive模式下需要精确计算池化窗口实际覆盖的有效元素个数(mean_matrix),然后将求和池化结果除以该值得到精确平均值。当前Ascend950缺少该算子的SIMT实现,导致包含AvgPool+exclusive的模型在950上无法正确执行。
算子计算公式:
y = x1 / mean_matrixmean_matrix[h,w] = mean_h * mean_w其中x1为求和池化输出,x2为原始输入feature map(仅用于获取输入空间尺寸,不参与数值计算)。
Origin(信息来源)
AscendC SIMT算子开发团队需求,基于TensorFlow AvgPool算子的exclusive模式语义,在Ascend950上实现SIMT版本的AvgPoolUpdate辅助算子。
Benefit / Necessity(价值/作用)
Design(设计方案)