diff --git a/paddle/fluid/inference/tests/api/trt_dynamic_shape_ernie_test.cc b/paddle/fluid/inference/tests/api/trt_dynamic_shape_ernie_test.cc
index e851be755855bc9ae5dc605c7e6b06602d64b959..870c73fe1e6f66dda324967f285c5dacc31ac073 100644
--- a/paddle/fluid/inference/tests/api/trt_dynamic_shape_ernie_test.cc
+++ b/paddle/fluid/inference/tests/api/trt_dynamic_shape_ernie_test.cc
@@ -379,7 +379,7 @@ void run(paddle_infer::Predictor* predictor, std::vector<float>* out_data) {
   // max_seq_len represents the max sentence length of all the sentences, only
   // length of
   // input i4 is useful, data means nothing.
-  int32_t i4[max_seq_len] = {0};
+  float i4[max_seq_len] = {0};
 
   auto input_names = predictor->GetInputNames();
   // first input