disable mkldnn fuse on ocr test

7 years ago · 26fc698f85
parent 1a99302c14
commit 26fc698f85
1 changed files with 5 additions and 0 deletions
--- a/paddle/fluid/inference/tests/api/analyzer_vis_tester.cc
+++ b/paddle/fluid/inference/tests/api/analyzer_vis_tester.cc
@ -62,7 +62,12 @@ void TestVisualPrediction(bool use_mkldnn) {
  cfg._use_mkldnn = use_mkldnn;
  cfg.device = 0;
  cfg.enable_ir_optim = true;
+  // TODO(TJ): fix fusion gru
  cfg.ir_passes.push_back("fc_gru_fuse_pass");
+#ifdef PADDLE_WITH_MKLDNN
+  // disable mkldnn fuse since it should have some bugs
+  cfg.ir_passes.push_back("conv_relu_mkldnn_fuse_pass");
+#endif
  predictor =
      CreatePaddlePredictor<AnalysisConfig, PaddleEngineKind::kAnalysis>(cfg);