Skip to content

Commit d6ff006

Browse files
committed
add serial to trt test and do not print log for unused trt logs
1 parent 0b38822 commit d6ff006

File tree

14 files changed

+16
-17
lines changed

14 files changed

+16
-17
lines changed

paddle/fluid/inference/analysis/data_flow_graph.cc

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -112,8 +112,8 @@ void DataFlowGraph::Build(const framework::proto::ProgramDesc &prog) {
112112
out_alias->SetPbMsg(out->pb_msg());
113113
var2id[out_alias->name()] =
114114
out_alias->id(); // update variable's alias Node
115-
LOG(INFO) << "loop found in graph, create SSA alias node ["
116-
<< out_alias->repr() << "] for [" << out->repr() << "]";
115+
VLOG(40) << "loop found in graph, create SSA alias node ["
116+
<< out_alias->repr() << "] for [" << out->repr() << "]";
117117
out = out_alias;
118118
}
119119
out->inlinks.push_back(o);

paddle/fluid/inference/tensorrt/convert/activation_op.cc

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -27,7 +27,7 @@ class ActivationOpConverter : public OpConverter {
2727
// Here the two nullptr looks strange, that's because the
2828
// framework::OpDesc's constructor is strange.
2929
framework::OpDesc op_desc(op, nullptr);
30-
LOG(INFO)
30+
VLOG(3)
3131
<< "convert a fluid Activation op to tensorrt activation layer whose "
3232
"type is "
3333
<< op_type_;

paddle/fluid/inference/tensorrt/convert/batch_norm_op.cc

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -23,7 +23,7 @@ class BatchNormOpConverter : public OpConverter {
2323
public:
2424
void operator()(const framework::proto::OpDesc& op,
2525
const framework::Scope& scope, bool test_mode) override {
26-
LOG(INFO) << "convert a fluid batch norm op to tensorrt batch_norm";
26+
VLOG(3) << "convert a fluid batch norm op to tensorrt batch_norm";
2727

2828
framework::OpDesc op_desc(op, nullptr);
2929
PADDLE_ENFORCE_EQ(op_desc.Input("X").size(), 1);

paddle/fluid/inference/tensorrt/convert/concat_op.cc

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -25,7 +25,7 @@ class ConcatOpConverter : public OpConverter {
2525
public:
2626
void operator()(const framework::proto::OpDesc& op,
2727
const framework::Scope& scope, bool test_mode) override {
28-
VLOG(40) << "convert a fluid mul op to tensorrt mul layer without bias";
28+
VLOG(3) << "convert a fluid mul op to tensorrt mul layer without bias";
2929

3030
framework::OpDesc op_desc(op, nullptr);
3131
// Declare inputs

paddle/fluid/inference/tensorrt/convert/conv2d_op.cc

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -37,8 +37,7 @@ class Conv2dOpConverter : public OpConverter {
3737
public:
3838
void operator()(const framework::proto::OpDesc& op,
3939
const framework::Scope& scope, bool test_mode) override {
40-
LOG(INFO)
41-
<< "convert a fluid conv2d op to tensorrt conv layer without bias";
40+
VLOG(3) << "convert a fluid conv2d op to tensorrt conv layer without bias";
4241

4342
framework::OpDesc op_desc(op, nullptr);
4443
PADDLE_ENFORCE_EQ(op_desc.Input("Input").size(), 1);

paddle/fluid/inference/tensorrt/convert/dropout_op.cc

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -25,7 +25,7 @@ class DropoutOpConverter : public OpConverter {
2525
public:
2626
void operator()(const framework::proto::OpDesc& op,
2727
const framework::Scope& scope, bool test_mode) override {
28-
VLOG(40) << "convert a fluid dropout op to tensorrt dropout layer";
28+
VLOG(3) << "convert a fluid dropout op to tensorrt dropout layer";
2929
framework::OpDesc op_desc(op, nullptr);
3030
// Declare inputs
3131
auto* input1 = engine_->GetITensor(op_desc.Input("X")[0]);

paddle/fluid/inference/tensorrt/convert/elementwise_op.cc

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -26,7 +26,7 @@ class ElementwiseWeightOpConverter : public OpConverter {
2626
// Here the two nullptr looks strange, that's because the
2727
// framework::OpDesc's constructor is strange.
2828
framework::OpDesc op_desc(op, nullptr);
29-
LOG(INFO) << "convert a fluid elementwise op to tensorrt IScaleLayer";
29+
VLOG(3) << "convert a fluid elementwise op to tensorrt IScaleLayer";
3030

3131
PADDLE_ENFORCE_EQ(op_desc.Input("X").size(), 1);
3232
PADDLE_ENFORCE_EQ(op_desc.Input("Y").size(), 1); // Y is a weight
@@ -108,7 +108,7 @@ class ElementwiseTensorOpConverter : public OpConverter {
108108
// Here the two nullptr looks strange, that's because the
109109
// framework::OpDesc's constructor is strange.
110110
framework::OpDesc op_desc(op, nullptr);
111-
LOG(INFO) << "convert a fluid elementwise op to tensorrt IScaleLayer";
111+
VLOG(3) << "convert a fluid elementwise op to tensorrt IScaleLayer";
112112

113113
PADDLE_ENFORCE_EQ(op_desc.Input("X").size(), 1);
114114
PADDLE_ENFORCE_EQ(op_desc.Input("Y").size(), 1); // Y is a weight

paddle/fluid/inference/tensorrt/convert/fc_op.cc

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -52,7 +52,7 @@ class FcOpConverter : public OpConverter {
5252
public:
5353
void operator()(const framework::proto::OpDesc& op,
5454
const framework::Scope& scope, bool test_mode) override {
55-
VLOG(40) << "convert a fluid fc op to tensorrt fc layer without bias";
55+
VLOG(3) << "convert a fluid fc op to tensorrt fc layer without bias";
5656

5757
framework::OpDesc op_desc(op, nullptr);
5858
PADDLE_ENFORCE_EQ(op_desc.Input("X").size(), 1);

paddle/fluid/inference/tensorrt/convert/mul_op.cc

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -25,7 +25,7 @@ class MulOpConverter : public OpConverter {
2525
public:
2626
void operator()(const framework::proto::OpDesc& op,
2727
const framework::Scope& scope, bool test_mode) override {
28-
VLOG(40) << "convert a fluid mul op to tensorrt mul layer without bias";
28+
VLOG(3) << "convert a fluid mul op to tensorrt mul layer without bias";
2929

3030
framework::OpDesc op_desc(op, nullptr);
3131
// Declare inputs

paddle/fluid/inference/tensorrt/convert/pad_op.cc

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -25,7 +25,7 @@ class PadOpConverter : public OpConverter {
2525
public:
2626
void operator()(const framework::proto::OpDesc& op,
2727
const framework::Scope& scope, bool test_mode) override {
28-
VLOG(40) << "convert a fluid transpose op to tensorrt tranpose layer";
28+
VLOG(3) << "convert a fluid transpose op to tensorrt tranpose layer";
2929

3030
framework::OpDesc op_desc(op, nullptr);
3131
// Declare inputs

0 commit comments

Comments
 (0)