From 3a0caf801fbf499216b92e32dc63bae4260dd8b6 Mon Sep 17 00:00:00 2001
From: nhzlx <zlx_hg@163.com>
Date: Wed, 8 Aug 2018 03:32:12 +0000
Subject: [PATCH] modify trt engine op test

---
 paddle/fluid/operators/tensorrt_engine_op_test.cc | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/paddle/fluid/operators/tensorrt_engine_op_test.cc b/paddle/fluid/operators/tensorrt_engine_op_test.cc
index b6e285dc5dc..37657fa0b04 100644
--- a/paddle/fluid/operators/tensorrt_engine_op_test.cc
+++ b/paddle/fluid/operators/tensorrt_engine_op_test.cc
@@ -103,7 +103,8 @@ TEST(TensorRTEngineOp, manual) {
   SetAttr<std::string>(engine_op_desc.Proto(), "engine_uniq_key", "a_engine");
   SetAttr<std::vector<std::string>>(engine_op_desc.Proto(), "parameters",
                                     std::vector<std::string>({}));
-  SetAttr<std::vector<std::string>>(engine_op_desc.Proto(), "output_name_mapping",
+  SetAttr<std::vector<std::string>>(engine_op_desc.Proto(),
+                                    "output_name_mapping",
                                     std::vector<std::string>({"z0"}));
 
   LOG(INFO) << "create engine op";
@@ -198,7 +199,8 @@ void Execute(int batch_size, int input_dim, int output_dim, int nlayers = 1) {
       std::vector<std::string>({"y0", "y1", "y2", "y3"}));
   SetAttr<std::string>(engine_op_desc.Proto(), "engine_uniq_key", "b_engine");
 
-  SetAttr<std::vector<std::string>>(engine_op_desc.Proto(), "output_name_mapping",
+  SetAttr<std::vector<std::string>>(engine_op_desc.Proto(),
+                                    "output_name_mapping",
                                     std::vector<std::string>({"z3"}));
 
   auto engine_op = framework::OpRegistry::CreateOp(*engine_op_desc.Proto());
-- 
GitLab