提交 26fc698f 编写于 作者: T tensor-tang

disable mkldnn fuse on ocr test

上级 1a99302c
...@@ -62,7 +62,12 @@ void TestVisualPrediction(bool use_mkldnn) { ...@@ -62,7 +62,12 @@ void TestVisualPrediction(bool use_mkldnn) {
cfg._use_mkldnn = use_mkldnn; cfg._use_mkldnn = use_mkldnn;
cfg.device = 0; cfg.device = 0;
cfg.enable_ir_optim = true; cfg.enable_ir_optim = true;
// TODO(TJ): fix fusion gru
cfg.ir_passes.push_back("fc_gru_fuse_pass"); 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 = predictor =
CreatePaddlePredictor<AnalysisConfig, PaddleEngineKind::kAnalysis>(cfg); CreatePaddlePredictor<AnalysisConfig, PaddleEngineKind::kAnalysis>(cfg);
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册