File tree Expand file tree Collapse file tree 1 file changed +1
-7
lines changed
paddle/fluid/inference/api/demo_ci Expand file tree Collapse file tree 1 file changed +1
-7
lines changed Original file line number Diff line number Diff line change @@ -18,15 +18,9 @@ limitations under the License. */
18
18
19
19
#include < gflags/gflags.h>
20
20
#include < glog/logging.h> // use glog instead of CHECK to avoid importing other paddle header files.
21
- #include < fstream>
22
- #include < iostream>
23
-
24
- // #include "paddle/fluid/platform/enforce.h"
25
21
#include " paddle/fluid/inference/demo_ci/utils.h"
26
22
27
- #ifdef PADDLE_WITH_CUDA
28
23
DECLARE_double (fraction_of_gpu_memory_to_use);
29
- #endif
30
24
DEFINE_string (modeldir, " " , " Directory of the inference model." );
31
25
DEFINE_string (refer, " " , " path to reference result for comparison." );
32
26
DEFINE_string (
@@ -38,7 +32,7 @@ namespace paddle {
38
32
namespace demo {
39
33
40
34
/*
41
- * Use the native fluid engine to inference the demo.
35
+ * Use the tensorrt fluid engine to inference the demo.
42
36
*/
43
37
void Main () {
44
38
std::unique_ptr<PaddlePredictor> predictor;
You can’t perform that action at this time.
0 commit comments