BatchNormV3在Ascend950路径上已有NCHW、NCDHW、NHWC、NDHWC等格式配置,但ND格式输入在arch35/regbase tiling中会被判为非法格式,导致aclnnBatchNorm在ND输入场景下无法完成workspace查询和kernel执行。
支持BatchNormV3在Ascend950上的FP32 ND格式训练场景:
ttk_batch_norm_v3_nd.csv
https://gitcode.com/cann/ops-nn/pulls/4921
需求背景
BatchNormV3在Ascend950路径上已有NCHW、NCDHW、NHWC、NDHWC等格式配置,但ND格式输入在arch35/regbase tiling中会被判为非法格式,导致aclnnBatchNorm在ND输入场景下无法完成workspace查询和kernel执行。
需求内容
支持BatchNormV3在Ascend950上的FP32 ND格式训练场景:
验证方式
ttk_batch_norm_v3_nd.csv,golden 100%,precision PASS。关联PR
https://gitcode.com/cann/ops-nn/pulls/4921