File tree Expand file tree Collapse file tree 2 files changed +6
-8
lines changed
paddle/fluid/inference/api/demo_ci Expand file tree Collapse file tree 2 files changed +6
-8
lines changed Original file line number Diff line number Diff line change @@ -62,10 +62,10 @@ include_directories("${PADDLE_LIB}/third_party/boost")
62
62
include_directories ("${PADDLE_LIB} /third_party/eigen3" )
63
63
64
64
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 ()
69
69
endif (NOT WIN32 )
70
70
71
71
if (NOT WIN32 )
Original file line number Diff line number Diff line change @@ -124,9 +124,7 @@ void Main(bool use_gpu, bool use_trt) {
124
124
config.device = 0 ;
125
125
config.max_batch_size = 1 ;
126
126
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);
130
128
}
131
129
132
130
VLOG (3 ) << " begin to process data" ;
@@ -166,7 +164,7 @@ int main(int argc, char** argv) {
166
164
} else if (FLAGS_use_gpu) {
167
165
paddle::demo::Main (true /* use_gpu*/ , false );
168
166
} else {
169
- paddle::demo::Main (false /* use_gpu*/ , false /* use_tensorrt*/ );
167
+ paddle::demo::Main (false /* use_gpu*/ , false /* use_tensorrt*/ );
170
168
}
171
169
return 0 ;
172
170
}
You can’t perform that action at this time.
0 commit comments