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 @@ -38,8 +38,8 @@ if(WITH_TESTING)
38
38
ARGS --word2vec_dirname=${WORD2VEC_MODEL_DIR} --book_dirname=${PYTHON_TESTS_DIR}/book )
39
39
set_tests_properties (test_api_impl PROPERTIES DEPENDS test_image_classification )
40
40
endif ()
41
- cc_test (test_analysis_predictor SRCS analysis_predictor_tester.cc DEPS analysis_predictor ${inference_deps} paddle_inference_api
42
- ARGS --dirname=${PYTHON_TESTS_DIR}/book )
41
+ cc_test (test_analysis_predictor SRCS analysis_predictor_tester.cc DEPS analysis_predictor ${inference_deps}
42
+ ARGS --dirname=${WORD2VEC_MODEL_DIR} )
43
43
44
44
if (WITH_GPU AND TENSORRT_FOUND )
45
45
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