diff --git a/paddle/fluid/inference/tensorrt/convert/test_elementwise_op.cc b/paddle/fluid/inference/tensorrt/convert/test_elementwise_op.cc
index c0254c61629f0b96bd2625a56db7974fda53f639..7537d02a35b66a41c158cd8eb1b1e5d4107e7d84 100644
--- a/paddle/fluid/inference/tensorrt/convert/test_elementwise_op.cc
+++ b/paddle/fluid/inference/tensorrt/convert/test_elementwise_op.cc
@@ -23,7 +23,7 @@ namespace tensorrt {
 TEST(elementwise_op, add_weight_test) {
   std::unordered_set<std::string> parameters({"elementwise_add-Y"});
   framework::Scope scope;
-  TRTConvertValidation validator(1, parameters, scope, 1 << 15);
+  TRTConvertValidation validator(10, parameters, scope, 1 << 15);
   validator.DeclInputVar("elementwise_add-X", nvinfer1::DimsCHW(10, 3, 3));
   validator.DeclParamVar("elementwise_add-Y", nvinfer1::Dims3(10, 1, 1));
   // validator.DeclParamVar("mul-Y", nvinfer1::Dims2(8, 2));
@@ -41,13 +41,13 @@ TEST(elementwise_op, add_weight_test) {
 
   validator.SetOp(*desc.Proto());
 
-  validator.Execute(1);
+  validator.Execute(8);
 }
 
 TEST(elementwise_op, add_tensor_test) {
   std::unordered_set<std::string> parameters;
   framework::Scope scope;
-  TRTConvertValidation validator(2, parameters, scope, 1 << 15);
+  TRTConvertValidation validator(8, parameters, scope, 1 << 15);
   validator.DeclInputVar("elementwise_add-X", nvinfer1::DimsCHW(10, 3, 3));
   validator.DeclInputVar("elementwise_add-Y", nvinfer1::Dims3(10, 3, 3));
   // validator.DeclParamVar("mul-Y", nvinfer1::Dims2(8, 2));
@@ -64,7 +64,7 @@ TEST(elementwise_op, add_tensor_test) {
 
   validator.SetOp(*desc.Proto());
 
-  validator.Execute(1);
+  validator.Execute(8);
 }
 
 }  // namespace tensorrt
diff --git a/paddle/fluid/inference/tensorrt/convert/ut_helper.h b/paddle/fluid/inference/tensorrt/convert/ut_helper.h
index 39529cc2c799212f91107b1b86dd2c8c3642b6da..63c2f978f253df11100ecca83acae5eab6a0337d 100644
--- a/paddle/fluid/inference/tensorrt/convert/ut_helper.h
+++ b/paddle/fluid/inference/tensorrt/convert/ut_helper.h
@@ -149,7 +149,7 @@ class TRTConvertValidation {
     cudaStreamSynchronize(*engine_->stream());
 
     ASSERT_FALSE(op_desc_->OutputArgumentNames().empty());
-    const size_t output_space_size = 2000;
+    const size_t output_space_size = 3000;
     for (const auto& output : op_desc_->OutputArgumentNames()) {
       std::vector<float> fluid_out;
       std::vector<float> trt_out(output_space_size);