Skip to content

Commit 387610a

Browse files
authored
Merge pull request #14325 from luotao1/fix_test_analysis_predictor
fix test_analysis_predictor
2 parents b03a44e + 381bea0 commit 387610a

File tree

2 files changed

+3
-3
lines changed

2 files changed

+3
-3
lines changed

paddle/fluid/inference/api/CMakeLists.txt

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -38,8 +38,8 @@ if(WITH_TESTING)
3838
ARGS --word2vec_dirname=${WORD2VEC_MODEL_DIR} --book_dirname=${PYTHON_TESTS_DIR}/book)
3939
set_tests_properties(test_api_impl PROPERTIES DEPENDS test_image_classification)
4040
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})
4343

4444
if(WITH_GPU AND TENSORRT_FOUND)
4545
cc_library(paddle_inference_tensorrt_subgraph_engine

paddle/fluid/inference/api/analysis_predictor_tester.cc

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -24,7 +24,7 @@ using contrib::AnalysisConfig;
2424

2525
TEST(AnalysisPredictor, ZeroCopy) {
2626
AnalysisConfig config;
27-
config.model_dir = FLAGS_dirname + "/word2vec.inference.model";
27+
config.model_dir = FLAGS_dirname;
2828
config.use_feed_fetch_ops = false;
2929

3030
auto predictor = CreatePaddlePredictor<AnalysisConfig>(config);

0 commit comments

Comments
 (0)