From 91e8fbac2fee6f03725eacad0f1b1c6ec2ade0df Mon Sep 17 00:00:00 2001
From: Michal Gallus <michal.gallus@intel.com>
Date: Fri, 12 Oct 2018 13:36:29 +0200
Subject: [PATCH] Enable MKLDNN in Resnet50Tester

test=develop
---
 paddle/fluid/inference/tests/api/analyzer_resnet50_tester.cc | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/paddle/fluid/inference/tests/api/analyzer_resnet50_tester.cc b/paddle/fluid/inference/tests/api/analyzer_resnet50_tester.cc
index 67668298440..49895bd7fc2 100644
--- 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) {
-- 
GitLab