已关闭
[ascend350] layer_norm等16个算子不支持ascend350平台编译 #5744
zhuzixian-lr创建于  10 天前关闭于  6 天前
zhuzixian-lr成员
10 天前 创建

问题描述

Ascend350 与 Ascend950 同为 DAV_3510 架构(GetCurNpuArch 均返回 NpuArch::DAV_3510,走同一代码分支),但以下 16 个算子未注册 ascend350,build.sh --soc=ascend350 构建时日志报 On [ascend350], [<op>] not supported.,无二进制产物:

layer_norm、layer_norm_v3、layer_norm_v4、layer_norm_grad_v3(含 LayerNormBetaGammaBackpropV2/LayerNormXBackpropV2 入口)、rms_norm、rms_norm_grad、add_rms_norm、inplace_add_rms_norm、add_layer_norm、lp_norm_v2、bn_training_reduce、broadcast_gradient_args、log_softmax_v2、softmax_grad、log_softmax_grad、confusion_softmax_grad

期望行为

上述算子在 ascend350 上与 ascend950 同配置可用,--soc=ascend350 可正常编出二进制。

复现步骤

bash build.sh --pkg --soc=ascend350 --ops=layer_norm --vendor_name=customize -j4
# 日志:On [ascend350], [layer_norm] not supported.
likedislike
yuning_chenyuning_chen成员
10 天前 将 zhuzixian-lr 设为负责人
CANN-robotCANN-robot成员
6 天前 关闭了 issue
CANN-robotCANN-robot成员
6 天前 添加了label:resolved