|
|
|
@ -226,21 +226,18 @@ void AnalysisPredictor::OptimizeInferenceProgram() {
|
|
|
|
|
argument_.origin_program_desc.reset(
|
|
|
|
|
new ProgramDesc(*inference_program_->Proto()));
|
|
|
|
|
|
|
|
|
|
bool use_mkldnn = config_._use_mkldnn;
|
|
|
|
|
switch (config_.ir_mode) {
|
|
|
|
|
case contrib::AnalysisConfig::IrPassMode::kExclude:
|
|
|
|
|
Analyzer()
|
|
|
|
|
.IncludeAllIrPasses()
|
|
|
|
|
.SetUseMkldnn(use_mkldnn)
|
|
|
|
|
.DisableIrPasses(use_mkldnn ? config_.ir_mkldnn_passes
|
|
|
|
|
: config_.ir_passes)
|
|
|
|
|
.SetUseMkldnn(config_._use_mkldnn)
|
|
|
|
|
.DisableIrPasses(config_.ir_passes)
|
|
|
|
|
.Run(&argument_);
|
|
|
|
|
break;
|
|
|
|
|
case contrib::AnalysisConfig::IrPassMode::kInclude:
|
|
|
|
|
Analyzer()
|
|
|
|
|
.SetUseMkldnn(use_mkldnn)
|
|
|
|
|
.IncludeIrPasses(use_mkldnn ? config_.ir_mkldnn_passes
|
|
|
|
|
: config_.ir_passes)
|
|
|
|
|
.SetUseMkldnn(config_._use_mkldnn)
|
|
|
|
|
.IncludeIrPasses(config_.ir_passes)
|
|
|
|
|
.Run(&argument_);
|
|
|
|
|
break;
|
|
|
|
|
default:
|
|
|
|
|