Skip to content

Commit 3a0caf8

Browse files
committed
modify trt engine op test
1 parent e51d045 commit 3a0caf8

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

paddle/fluid/operators/tensorrt_engine_op_test.cc

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -103,7 +103,8 @@ TEST(TensorRTEngineOp, manual) {
103103
SetAttr<std::string>(engine_op_desc.Proto(), "engine_uniq_key", "a_engine");
104104
SetAttr<std::vector<std::string>>(engine_op_desc.Proto(), "parameters",
105105
std::vector<std::string>({}));
106-
SetAttr<std::vector<std::string>>(engine_op_desc.Proto(), "output_name_mapping",
106+
SetAttr<std::vector<std::string>>(engine_op_desc.Proto(),
107+
"output_name_mapping",
107108
std::vector<std::string>({"z0"}));
108109

109110
LOG(INFO) << "create engine op";
@@ -198,7 +199,8 @@ void Execute(int batch_size, int input_dim, int output_dim, int nlayers = 1) {
198199
std::vector<std::string>({"y0", "y1", "y2", "y3"}));
199200
SetAttr<std::string>(engine_op_desc.Proto(), "engine_uniq_key", "b_engine");
200201

201-
SetAttr<std::vector<std::string>>(engine_op_desc.Proto(), "output_name_mapping",
202+
SetAttr<std::vector<std::string>>(engine_op_desc.Proto(),
203+
"output_name_mapping",
202204
std::vector<std::string>({"z3"}));
203205

204206
auto engine_op = framework::OpRegistry::CreateOp(*engine_op_desc.Proto());

0 commit comments

Comments
 (0)