From 25ca2ca00107b83fb68655a21ac37ccf808e8987 Mon Sep 17 00:00:00 2001
From: Tao Luo <luotao02@baidu.com>
Date: Fri, 8 Mar 2019 11:40:41 +0800
Subject: [PATCH] change init_idx to INT32 in transformer_test

test=develop
---
 paddle/fluid/inference/tests/api/analyzer_transformer_tester.cc | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/paddle/fluid/inference/tests/api/analyzer_transformer_tester.cc b/paddle/fluid/inference/tests/api/analyzer_transformer_tester.cc
index d3f97f2379..9d17f38ab7 100644
--- a/paddle/fluid/inference/tests/api/analyzer_transformer_tester.cc
+++ b/paddle/fluid/inference/tests/api/analyzer_transformer_tester.cc
@@ -147,7 +147,7 @@ void PrepareInputs(std::vector<PaddleTensor> *input_slots, DataRecord *data,
 
   init_idx.name = "init_idx";
   init_idx.shape.assign({batch_size});
-  init_idx.dtype = PaddleDType::INT64;
+  init_idx.dtype = PaddleDType::INT32;
   TensorAssignData<int64_t>(&init_idx, one_batch.init_idx);
 
   trg_src_attn_bias.name = "trg_src_attn_bias";
-- 
GitLab