xxq250
/
mindspore
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

!20413 [MS][LITE]return error when subgraph kernel is nullptr 

Merge pull request !20413 from mengyuanli/bugfix
This commit is contained in:
i-robot 2021-07-19 01:09:21 +00:00 committed by Gitee
parent a455d81133 04f833ec5e
commit c9af7643d7
3 changed files with 6 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
mindspore/lite/src/scheduler.cc
View File

@ -1050,6 +1050,10 @@ std::vector<kernel::LiteKernel *> Scheduler::ScheduleSubGraphToSubGraphKernels(c
return ScheduleMainSubGraphToKernels();
}
auto subgraph_kernel = SchedulePartialToSubGraphKernel(subgraph_index);
if (subgraph_kernel == nullptr) {
MS_LOG(ERROR) << "SchedulePartialToSubGraphKernel failed, subgraph_index: " << subgraph_index;
return {};
}
subgraph_kernel->set_name("subgraph_" + std::to_string(subgraph_index));
subgraph_index_subgraph_kernel_map_[subgraph_index] = subgraph_kernel;
return {subgraph_kernel};

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

@ -21,7 +21,6 @@ mtk_transformer_decoder_joint.tflite
quant_aware_bank_card_detection_inception.onnx
quant_aware_bank_card_recognition_fcny.onnx
quant_aware_identify_card_detect.onnx
tiny-yolov3-11.onnx;2;1,416,416,3:1,2
# cur acc for ml_video_edit_art_transfer is 2+%
ml_video_edit_art_transfer.onnx;3
#ml_table_detection.onnx: onnx quantized model

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

@ -98,3 +98,5 @@ ml_audio_kit_vocals_test.onnx;1;1,512,1024,2 2
gender_lstm_scd.onnx
gender_lstm_vad.onnx
gender_resnet34_lzl.onnx
# cur acc for tiny-yolov3-11 is 2.5% because the Unsqueeze_concat_7:0's output of the last op has very small numbers.
tiny-yolov3-11.onnx;2;1,224,224,3:1,2 3