Enable MKLDNN in Resnet50Tester

test=develop
6 years ago · 91e8fbac2f
parent 582f59c190
commit 91e8fbac2f
1 changed files with 3 additions and 0 deletions
--- a/paddle/fluid/inference/tests/api/analyzer_resnet50_tester.cc
+++ b/paddle/fluid/inference/tests/api/analyzer_resnet50_tester.cc
@ -27,6 +27,9 @@ void SetConfig(AnalysisConfig *cfg) {
  cfg->device = 0;
  cfg->enable_ir_optim = true;
  cfg->specify_input_name = true;
+#ifdef PADDLE_WITH_MKLDNN
+  cfg->_use_mkldnn = true;
+#endif
 }

 void SetInput(std::vector<std::vector<PaddleTensor>> *inputs) {