File tree Expand file tree Collapse file tree 2 files changed +4
-10
lines changed
paddle/fluid/inference/api Expand file tree Collapse file tree 2 files changed +4
-10
lines changed Original file line number Diff line number Diff line change @@ -226,21 +226,18 @@ void AnalysisPredictor::OptimizeInferenceProgram() {
226
226
argument_.origin_program_desc .reset (
227
227
new ProgramDesc (*inference_program_->Proto ()));
228
228
229
- bool use_mkldnn = config_._use_mkldnn ;
230
229
switch (config_.ir_mode ) {
231
230
case contrib::AnalysisConfig::IrPassMode::kExclude :
232
231
Analyzer ()
233
232
.IncludeAllIrPasses ()
234
- .SetUseMkldnn (use_mkldnn)
235
- .DisableIrPasses (use_mkldnn ? config_.ir_mkldnn_passes
236
- : config_.ir_passes )
233
+ .SetUseMkldnn (config_._use_mkldnn )
234
+ .DisableIrPasses (config_.ir_passes )
237
235
.Run (&argument_);
238
236
break ;
239
237
case contrib::AnalysisConfig::IrPassMode::kInclude :
240
238
Analyzer ()
241
- .SetUseMkldnn (use_mkldnn)
242
- .IncludeIrPasses (use_mkldnn ? config_.ir_mkldnn_passes
243
- : config_.ir_passes )
239
+ .SetUseMkldnn (config_._use_mkldnn )
240
+ .IncludeIrPasses (config_.ir_passes )
244
241
.Run (&argument_);
245
242
break ;
246
243
default :
Original file line number Diff line number Diff line change @@ -262,7 +262,6 @@ struct AnalysisConfig : public NativeConfig {
262
262
void SetIncludeMode () {
263
263
ir_mode = IrPassMode::kInclude ;
264
264
ir_passes = {" infer_clean_graph_pass" };
265
- ir_mkldnn_passes = {" infer_clean_graph_pass" };
266
265
}
267
266
268
267
// Determine whether to perform graph optimization.
@@ -271,8 +270,6 @@ struct AnalysisConfig : public NativeConfig {
271
270
IrPassMode ir_mode{IrPassMode::kExclude };
272
271
// passes to be excluded/included
273
272
std::vector<std::string> ir_passes{" embedding_fc_lstm_fuse_pass" };
274
- // passes to be excluded/included when MKL-DNN is enabled
275
- std::vector<std::string> ir_mkldnn_passes{" embedding_fc_lstm_fuse_pass" };
276
273
277
274
// NOT stable yet.
278
275
bool use_feed_fetch_ops{true };
You can’t perform that action at this time.
0 commit comments