Enable MKLDNN in Resnet50Tester

test=develop

Enable MKLDNN in Resnet50Tester
test=develop
91e8fbac · Michal Gallus · 582f59c1 · 91e8fbac
隐藏空白更改
内联并排

Showing with 3 addition and 0 deletion

paddle/fluid/inference/tests/api/analyzer_resnet50_tester.cc paddle/fluid/inference/tests/api/analyzer_resnet50_tester.cc +3 -0

未找到文件。
--- 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) {