From 602e28bf1c30cd72e7378d6dc1071423086bdc73 Mon Sep 17 00:00:00 2001
From: tensor-tang <tangjian03@baidu.com>
Date: Fri, 25 May 2018 14:38:01 +0800
Subject: [PATCH] use the actual data

---
 .../tests/book/test_inference_nlp.cc          | 48 +++++++++----------
 1 file changed, 23 insertions(+), 25 deletions(-)

diff --git a/paddle/fluid/inference/tests/book/test_inference_nlp.cc b/paddle/fluid/inference/tests/book/test_inference_nlp.cc
index 0d6d0adfb23..27bdd5528ef 100644
--- a/paddle/fluid/inference/tests/book/test_inference_nlp.cc
+++ b/paddle/fluid/inference/tests/book/test_inference_nlp.cc
@@ -12,6 +12,8 @@ WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 See the License for the specific language governing permissions and
 limitations under the License. */
 
+#include <sys/time.h>
+#include <time.h>
 #include "gflags/gflags.h"
 #include "gtest/gtest.h"
 #include "paddle/fluid/inference/tests/test_helper.h"
@@ -29,32 +31,28 @@ TEST(inference, understand_sentiment) {
   // 0. Call `paddle::framework::InitDevices()` initialize all the devices
   // In unittests, this is done in paddle/testing/paddle_gtest_main.cc
   paddle::framework::LoDTensor words;
-
-  paddle::framework::LoD lod{{0, 83}};
-  int64_t word_dict_len = 198392;
-  SetupLoDTensor(&words, lod, static_cast<int64_t>(0),
-                 static_cast<int64_t>(word_dict_len - 1));
   /*
-    std::vector<int64_t> srcdata{
-        784,    784,   1550,   6463,   56,     75693, 6189,  784,    784,  1550,
-        198391, 6463,  42468,  4376,   10251,  10760, 6189,  297,    396,  6463,
-        6463,   1550,  198391, 6463,   22564,  1612,  291,   68,     164,  784,
-        784,    1550,  198391, 6463,   13659,  3362,  42468, 6189,   2209,
-    198391,
-        6463,   2209,  2209,   198391, 6463,   2209,  1062,  3029,   1831, 3029,
-        1065,   2281,  100,    11216,  1110,   56,    10869, 9811,   100,
-    198391,
-        6463,   100,   9280,   100,    288,    40031, 1680,  1335,   100,  1550,
-        9280,   7265,  244,    1550,   198391, 6463,  1550,  198391, 6463,
-    42468,
-        4376,   10251, 10760};
-    paddle::framework::LoD lod{{0, srcdata.size()}};
-    words.set_lod(lod);
-    int64_t* pdata =
-    words.mutable_data<int64_t>({static_cast<int64_t>(srcdata.size()), 1},
-                                                  paddle::platform::CPUPlace());
-    memcpy(pdata, srcdata.data(), words.numel() * sizeof(int64_t));
-  */
+    paddle::framework::LoD lod{{0, 83}};
+    int64_t word_dict_len = 198392;
+    SetupLoDTensor(&words, lod, static_cast<int64_t>(0),
+                   static_cast<int64_t>(word_dict_len - 1));
+   */
+  std::vector<int64_t> srcdata{
+      784,    784,   1550,   6463,   56,     75693, 6189,  784,    784,  1550,
+      198391, 6463,  42468,  4376,   10251,  10760, 6189,  297,    396,  6463,
+      6463,   1550,  198391, 6463,   22564,  1612,  291,   68,     164,  784,
+      784,    1550,  198391, 6463,   13659,  3362,  42468, 6189,   2209, 198391,
+      6463,   2209,  2209,   198391, 6463,   2209,  1062,  3029,   1831, 3029,
+      1065,   2281,  100,    11216,  1110,   56,    10869, 9811,   100,  198391,
+      6463,   100,   9280,   100,    288,    40031, 1680,  1335,   100,  1550,
+      9280,   7265,  244,    1550,   198391, 6463,  1550,  198391, 6463, 42468,
+      4376,   10251, 10760};
+  paddle::framework::LoD lod{{0, srcdata.size()}};
+  words.set_lod(lod);
+  int64_t* pdata = words.mutable_data<int64_t>(
+      {static_cast<int64_t>(srcdata.size()), 1}, paddle::platform::CPUPlace());
+  memcpy(pdata, srcdata.data(), words.numel() * sizeof(int64_t));
+
   LOG(INFO) << "number of input size:" << words.numel();
   std::vector<paddle::framework::LoDTensor*> cpu_feeds;
   cpu_feeds.push_back(&words);
-- 
GitLab