vilc
/
mindspore2022
forked from huawei/mindspore2022
0
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix infer bug of encoder_v5.pb at NPU device

This commit is contained in:
wangyanling 2021-08-02 19:54:34 +08:00
parent 7f71a99993
commit 060c4be964
6 changed files with 17 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
mindspore/lite/src/delegate/npu/npu_delegate.cc
View File

@ -206,6 +206,14 @@ int NPUDelegate::Build(DelegateModel *model) {
}
NPUOp *NPUDelegate::GetOP(kernel::Kernel *kernel, const schema::Primitive *primitive) {
if (primitive == nullptr) {
MS_LOG(ERROR) << "primitive is NULL!";
return nullptr;
}
if (kernel == nullptr) {
MS_LOG(ERROR) << "kernel is NULL!";
return nullptr;
}
auto name = kernel->name();
NPUOp *npu_op = nullptr;
auto node_type = primitive->value_type();

1
mindspore/lite/test/config/models_npu.cfg
View File

@ -88,3 +88,4 @@ ml_video_edit_hair_dyeing_segmodel_v2 0.5
ml_video_edit_makeup_mobilenetv203.onnx 2
ml_video_edit_hairline_segmentation;3 0.5
ml_video_edit_hair_dyeing_migrate_v2.onnx;4 0.5
ml_audio_kit_encoder_v5.pb;6;1,32:1,32:1,32:1,32:1:1

1
mindspore/lite/test/config/models_onnx.cfg
View File

@ -103,3 +103,4 @@ tiny-yolov3-11.onnx;2;1,224,224,3:1,2 3
# cur acc for ml_video_edit_art_transfer is 2+%
ml_video_edit_art_transfer.onnx;3
ssd-10.onnx;;;;calib_only
Q888_CV_face_recognition_self.onnx

1
mindspore/lite/test/config/models_onnx_fp16.cfg
View File

@ -102,3 +102,4 @@ ml_asr_decoder_202103.onnx;2;1,64,512:1,64 0.5
ml_video_edit_makeup_mobilenetv203.onnx 4
# The input of ml_video_edit_hair_dyeing_migrate_v2.onnx should be between [0, 1]
ml_video_edit_hair_dyeing_migrate_v2.onnx;4 2.5
Q888_CV_face_recognition_self.onnx 3.5

4
mindspore/lite/test/config/models_tf.cfg
View File

@ -104,4 +104,6 @@ hiai_nlu_model_v1.pb;3;1,16:1,16:1,16 2.0
hiai_nlu_model_v2.pb;7;1,5:1,6:1,174:1,98:1,5:1,5:1,5
hiai_nlu_model_multi.pb;6;1,32:1,32:1,6:1,11:1,74:1,32
hiai_nlu_model_single.pb;3;1,32:1,32:1,32
fsr_270_mindspore.pb
fsr_360_mindspore.pb
fsr_720_mindspore.pb

3
mindspore/lite/test/config/models_tf_fp16.cfg
View File

@ -88,3 +88,6 @@ hiai_transformer_encoder.pb;15 4
decoder_step_nocumsum_v5.pb;13;1:1,512:1,1429,2:1,127:1,127:1,127:1,127,320:1,80:1,512:1,512:1,512:1,512:1,512 1.2
hiai_nlu_model_multi.pb;6;1,32:1,32:1,6:1,11:1,74:1,32 25
hiai_nlu_model_single.pb;3;1,32:1,32:1,32 2470
fsr_270_mindspore.pb 6.0
fsr_360_mindspore.pb 6.5
fsr_720_mindspore.pb 2.0