已关闭
[Requirement|需求建议]: maxPoolV3Grad支持确定性实现 #4467
m0_46386992创建于  7月31日关闭于  8月10日
m0_46386992
m0_46386992成员
7月31日 创建

Thanks for sending an requirement! Please fill in the following template to help quickly solve your problem.

Backgroud(背景信息)

maxPoolV3Grad当前在重叠路径下采用非确定性实现,需要增加确定性模板实现。

Origin(信息来源)

vector

Benefit / Necessity (价值/作用)

Design(设计方案)

采用模板路径区分方法:

  1. 在非重叠场景下使用现有方案,按照输出窗口划分,反向找到输入位置并写入,性能好且省空间
  2. 在重叠场景下按照输入进行划分,先维护最大位置列表,然后再遍历每个输入元素对应的 覆盖该元素的窗口的最大位置,如果等同于当前输入,则进行累加。可避免非确定性问题。代价是会额外申请argmax内存空间 + 额外的forward遍历开销。
likedislike
m0_46386992m0_46386992成员
7月31日 添加了label:requirement
m0_46386992m0_46386992成员
7月31日 将 m0_46386992 设为负责人
m0_46386992m0_46386992成员
8月5日 关联了pull request:[CANNBot]MaxPoolV3Grad算子支持Ascend950
CANN-robotCANN-robot成员
8月10日 关闭了 issue
CANN-robotCANN-robot成员
8月10日 添加了label:resolved