Skip to content

Commit d347ea6

Browse files
committed
fix comments
1 parent f3af90d commit d347ea6

File tree

2 files changed

+6
-8
lines changed

2 files changed

+6
-8
lines changed

paddle/fluid/inference/api/demo_ci/CMakeLists.txt

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -62,10 +62,10 @@ include_directories("${PADDLE_LIB}/third_party/boost")
6262
include_directories("${PADDLE_LIB}/third_party/eigen3")
6363

6464
if (NOT WIN32)
65-
if (USE_TENSORRT AND WITH_GPU)
66-
include_directories("${TENSORRT_INCLUDE_DIR}")
67-
link_directories("${TENSORRT_LIB_DIR}")
68-
endif()
65+
if (USE_TENSORRT AND WITH_GPU)
66+
include_directories("${TENSORRT_INCLUDE_DIR}")
67+
link_directories("${TENSORRT_LIB_DIR}")
68+
endif()
6969
endif(NOT WIN32)
7070

7171
if (NOT WIN32)

paddle/fluid/inference/api/demo_ci/vis_demo.cc

Lines changed: 2 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -124,9 +124,7 @@ void Main(bool use_gpu, bool use_trt) {
124124
config.device = 0;
125125
config.max_batch_size = 1;
126126
config.fraction_of_gpu_memory = 0.1; // set by yourself
127-
predictor =
128-
CreatePaddlePredictor<paddle::contrib::MixedRTConfig,
129-
PaddleEngineKind::kAutoMixedTensorRT>(config);
127+
predictor = CreatePaddlePredictor<paddle::contrib::MixedRTConfig>(config);
130128
}
131129

132130
VLOG(3) << "begin to process data";
@@ -166,7 +164,7 @@ int main(int argc, char** argv) {
166164
} else if (FLAGS_use_gpu) {
167165
paddle::demo::Main(true /*use_gpu*/, false);
168166
} else {
169-
paddle::demo::Main(false /* use_gpu*/, false /*use_tensorrt*/);
167+
paddle::demo::Main(false /*use_gpu*/, false /*use_tensorrt*/);
170168
}
171169
return 0;
172170
}

0 commit comments

Comments
 (0)