From 6a32f19865ac7d1a7b439b609b757f6aa08baceb Mon Sep 17 00:00:00 2001 From: tensor-tang Date: Tue, 12 Jun 2018 16:23:33 +0800 Subject: [PATCH] fix unknown use_mkldnn --- paddle/fluid/inference/tests/book/test_inference_nlp.cc | 1 + 1 file changed, 1 insertion(+) diff --git a/paddle/fluid/inference/tests/book/test_inference_nlp.cc b/paddle/fluid/inference/tests/book/test_inference_nlp.cc index 9dcd79c3b..4dc576ced 100644 --- a/paddle/fluid/inference/tests/book/test_inference_nlp.cc +++ b/paddle/fluid/inference/tests/book/test_inference_nlp.cc @@ -29,6 +29,7 @@ DEFINE_string(data_file, "", "File of input index data."); DEFINE_int32(repeat, 100, "Running the inference program repeat times"); DEFINE_bool(prepare_vars, true, "Prepare variables before executor"); DEFINE_int32(num_threads, 1, "Number of threads should be used"); +DECLARE_bool(use_mkldnn); inline double GetCurrentMs() { struct timeval time; -- GitLab