File tree Expand file tree Collapse file tree 2 files changed +3
-3
lines changed
paddle/fluid/inference/api Expand file tree Collapse file tree 2 files changed +3
-3
lines changed Original file line number Diff line number Diff line change @@ -37,8 +37,8 @@ if(WITH_TESTING)
37
37
ARGS --word2vec_dirname=${WORD2VEC_MODEL_DIR} --book_dirname=${PYTHON_TESTS_DIR}/book )
38
38
set_tests_properties (test_api_impl PROPERTIES DEPENDS test_image_classification )
39
39
endif ()
40
- cc_test (test_analysis_predictor SRCS analysis_predictor_tester.cc DEPS analysis_predictor ${inference_deps} paddle_inference_api
41
- ARGS --dirname=${PYTHON_TESTS_DIR}/book )
40
+ cc_test (test_analysis_predictor SRCS analysis_predictor_tester.cc DEPS analysis_predictor ${inference_deps}
41
+ ARGS --dirname=${WORD2VEC_MODEL_DIR} )
42
42
43
43
if (WITH_GPU AND TENSORRT_FOUND )
44
44
cc_library (paddle_inference_tensorrt_subgraph_engine
Original file line number Diff line number Diff line change @@ -24,7 +24,7 @@ using contrib::AnalysisConfig;
24
24
25
25
TEST (AnalysisPredictor, ZeroCopy) {
26
26
AnalysisConfig config;
27
- config.model_dir = FLAGS_dirname + " /word2vec.inference.model " ;
27
+ config.model_dir = FLAGS_dirname;
28
28
config.use_feed_fetch_ops = false ;
29
29
30
30
auto predictor = CreatePaddlePredictor<AnalysisConfig>(config);
You can’t perform that action at this time.
0 commit comments