We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 1a99302 commit 26fc698Copy full SHA for 26fc698
paddle/fluid/inference/tests/api/analyzer_vis_tester.cc
@@ -62,7 +62,12 @@ void TestVisualPrediction(bool use_mkldnn) {
62
cfg._use_mkldnn = use_mkldnn;
63
cfg.device = 0;
64
cfg.enable_ir_optim = true;
65
+ // TODO(TJ): fix fusion gru
66
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
71
predictor =
72
CreatePaddlePredictor<AnalysisConfig, PaddleEngineKind::kAnalysis>(cfg);
73
0 commit comments