已关闭
[Requirement|需求建议]: max_pool3d_grad增加性能模板 #6242
liuchuangdev创建于  27 天前关闭于  26 天前
liuchuangdev成员
27 天前 创建

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

Backgroud(背景信息)

max_pool3d_grad(arch35)此前仅有 NCDHW 路径(SIMT 兜底):

  1. 无 NDHWC 原生路径:训练/部署中张量常为 NDHWC 布局,走 NCDHW 需额外转置,访存与向量化效率差;缺少 NDHWC 正/反向 kernel、tiling 与公共基类。
  2. NCDHW 大 kernel 问题:前向拷贝逐段下发、UB 预算未对齐封顶、NoSplit 判据口径不一致导致潜在越界、D 维搬运未合并、CalcRealIndex3D 参数序有误。
  3. 公共层缺失:pool_grad_common 无 NDHWC tiling 基类;设备侧 tiling 结构 Pool3DGradNCDHWTilingData 缺 NDHWC 变体与 isBigKernel/totalBufferSize 等字段。

Origin(信息来源)

950 vector

Benefit / Necessity (价值/作用)

Design(设计方案)

likedislike
Lliuchuangdev成员
27 天前 添加了label:requirement
林林泽辉成员
27 天前 关联了pull request:[WIP]einsum_pass图融合开源
林林泽辉成员
27 天前 删除了关联的pull request:[WIP]einsum_pass图融合开源
yuning_chenyuning_chen成员
27 天前 将 liuchuangdev 设为负责人
CANN-robotCANN-robot成员
26 天前 关闭了 issue
CANN-robotCANN-robot成员
26 天前 添加了label:resolved