From 26fc698f8510873594e7abbd9e64d141f1233887 Mon Sep 17 00:00:00 2001 From: tensor-tang Date: Fri, 14 Sep 2018 13:11:50 +0800 Subject: [PATCH] disable mkldnn fuse on ocr test --- paddle/fluid/inference/tests/api/analyzer_vis_tester.cc | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/paddle/fluid/inference/tests/api/analyzer_vis_tester.cc b/paddle/fluid/inference/tests/api/analyzer_vis_tester.cc index 0591869996e..a207c41b714 100644 --- 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(cfg); -- GitLab