已关闭
BatchNormV3支持Ascend950 ND格式 #2723
qianzehong创建于  5月16日关闭于  5月18日
qianzehong成员
5月16日 创建

需求背景

BatchNormV3在Ascend950路径上已有NCHW、NCDHW、NHWC、NDHWC等格式配置,但ND格式输入在arch35/regbase tiling中会被判为非法格式,导致aclnnBatchNorm在ND输入场景下无法完成workspace查询和kernel执行。

需求内容

支持BatchNormV3在Ascend950上的FP32 ND格式训练场景:

  • ND输入按BatchNorm通用语义解析,C维取dim1,归约维为dim0和dim2及后续维度。
  • 补齐Ascend950 FP32 ND OpDef和binary配置。
  • 新增ND格式tiling UT和TTK ACLNN用例。

验证方式

  • 本地ophost UT:40个用例通过。
  • 远端Ascend950构建:batch_norm_v3 binary/package构建通过。
  • 远端Ascend950 TTK:ttk_batch_norm_v3_nd.csv,golden 100%,precision PASS。

关联PR

https://gitcode.com/cann/ops-nn/pulls/4921

likedislike
oscillatedoscillated成员
5月16日 将 qianzehong 设为负责人
CANN-robotCANN-robot成员
5月18日 关闭了 issue
CANN-robotCANN-robot成员
5月18日 添加了label:resolved