Skip to content

Commit 26fc698

Browse files
committed
disable mkldnn fuse on ocr test
1 parent 1a99302 commit 26fc698

File tree

1 file changed

+5
-0
lines changed

1 file changed

+5
-0
lines changed

paddle/fluid/inference/tests/api/analyzer_vis_tester.cc

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -62,7 +62,12 @@ void TestVisualPrediction(bool use_mkldnn) {
6262
cfg._use_mkldnn = use_mkldnn;
6363
cfg.device = 0;
6464
cfg.enable_ir_optim = true;
65+
// TODO(TJ): fix fusion gru
6566
cfg.ir_passes.push_back("fc_gru_fuse_pass");
67+
#ifdef PADDLE_WITH_MKLDNN
68+
// disable mkldnn fuse since it should have some bugs
69+
cfg.ir_passes.push_back("conv_relu_mkldnn_fuse_pass");
70+
#endif
6671
predictor =
6772
CreatePaddlePredictor<AnalysisConfig, PaddleEngineKind::kAnalysis>(cfg);
6873

0 commit comments

Comments
 (0)