Ascend350 与 Ascend950 同为 DAV_3510 架构(GetCurNpuArch 均返回 NpuArch::DAV_3510,走同一代码分支),但以下 16 个算子未注册 ascend350,build.sh --soc=ascend350 构建时日志报 On [ascend350], [<op>] not supported.,无二进制产物:
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 可正常编出二进制。
--soc=ascend350
bash build.sh --pkg --soc=ascend350 --ops=layer_norm --vendor_name=customize -j4 # 日志:On [ascend350], [layer_norm] not supported.
问题描述
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.