问题现象:

EZ3002: [PID: 782638] 2024-12-13-16:19:39.794.329 Optype [Conv2DTranspose] of Ops kernel [AIcoreEngine] is unsupported.

Reason: [tbe-custot found in this op store.[tbe-custom]:op type Conv2DTranspose is not found in this op store.[Dynamic shape check]: data type DT_INT64 ofrted. All supported data type and format of tensor input0.input_size is: Data Type: {DT_INT32,DT_INT32}Format:{ND,ND}. 

排查过程:Conv2DTranspose对应的是onnx算子Convtranspose,查询CANN算子清单后得知Conv2DTranspose的input0不支持int64输出支持int32输入

问题原因:Conv2DTranspose不支持int64输出支持int32输入

解决方案:使用ATC参数--customize_dtypes将Conv2DTranspose的输入配置为int32的输入,

customize_dtypes.cfg配置:OpType::Conv2DTranspose:InputDtype:int32,float16,float16,float16,OutputDtype:float16
重新进行ATC转换
atc --model=modified_model_2.onnx --input_shape="phone:1,-1,768;phone_lengths:1;pitch:1,-1;pitchf:1,-1;ds" --output=ceshi --customize_dtypes=customize_dtypes.cfg --framework=5 --soc_version=Ascend310B4

Logo

鲲鹏昇腾开发者社区是面向全社会开放的“联接全球计算开发者,聚合华为+生态”的社区,内容涵盖鲲鹏、昇腾资源,帮助开发者快速获取所需的知识、经验、软件、工具、算力,支撑开发者易学、好用、成功,成为核心开发者。

更多推荐