diff --git a/paddle/fluid/inference/tests/api/analyzer_vis_tester.cc b/paddle/fluid/inference/tests/api/analyzer_vis_tester.cc index 0591869996e393ee3c701774dba86ff53b5e6e33..a207c41b7140c806b4c1fdc7f24a317b165c9aef 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);