Skip to content

Commit 6728d96

Browse files
committed
follow comments
1 parent 3de43a8 commit 6728d96

File tree

1 file changed

+0
-2
lines changed

1 file changed

+0
-2
lines changed

paddle/fluid/inference/tensorrt/test_engine.cc

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -82,7 +82,6 @@ TEST_F(TensorRTEngineTest, add_layer_multi_dim) {
8282
// It seems tensorrt FC use col-major: [[1.0, 3.3], [1.1, 4.4]]
8383
// instead of row-major, which is [[1.0, 1.1], [3.3, 4.4]]
8484
float raw_weight[4] = {1.0, 1.1, 3.3, 4.4};
85-
// [1, 2]
8685
float raw_bias[2] = {1.3, 2.4};
8786

8887
TensorRTEngine::Weight weight(nvinfer1::DataType::kFLOAT, raw_weight, 4);
@@ -97,7 +96,6 @@ TEST_F(TensorRTEngineTest, add_layer_multi_dim) {
9796
engine_->FreezeNetwork();
9897
ASSERT_EQ(engine_->engine()->getNbBindings(), 2);
9998

100-
// fill in real data [1.0, 2.0]
10199
float x_v[2] = {1.0, 2.0};
102100
engine_->SetInputFromCPU("x", reinterpret_cast<void*>(&x_v),
103101
2 * sizeof(float));

0 commit comments

Comments
 (0)