未验证 提交 75a13114 编写于 作者: T tianshuo78520a 提交者: GitHub

Fix inference CI bug (#26080)

* Fix inference bug

* fix inference lib
上级 34eb27a1
...@@ -1267,9 +1267,13 @@ EOF ...@@ -1267,9 +1267,13 @@ EOF
./run.sh ${PADDLE_ROOT} ${WITH_MKL:-ON} ${WITH_GPU:-OFF} ${INFERENCE_DEMO_INSTALL_DIR} \ ./run.sh ${PADDLE_ROOT} ${WITH_MKL:-ON} ${WITH_GPU:-OFF} ${INFERENCE_DEMO_INSTALL_DIR} \
${TENSORRT_INCLUDE_DIR:-/usr/local/TensorRT/include} \ ${TENSORRT_INCLUDE_DIR:-/usr/local/TensorRT/include} \
${TENSORRT_LIB_DIR:-/usr/local/TensorRT/lib} ${TENSORRT_LIB_DIR:-/usr/local/TensorRT/lib}
EXIT_CODE=$?
fluid_endTime_s=`date +%s` fluid_endTime_s=`date +%s`
echo "test_fluid_lib Total Time: $[ $fluid_endTime_s - $fluid_startTime_s ]s" echo "test_fluid_lib Total Time: $[ $fluid_endTime_s - $fluid_startTime_s ]s"
./clean.sh ./clean.sh
if [[ "$EXIT_CODE" != "0" ]]; then
exit 8;
fi
} }
function test_fluid_lib_train() { function test_fluid_lib_train() {
...@@ -1281,9 +1285,13 @@ EOF ...@@ -1281,9 +1285,13 @@ EOF
fluid_train_startTime_s=`date +%s` fluid_train_startTime_s=`date +%s`
cd ${PADDLE_ROOT}/paddle/fluid/train/demo cd ${PADDLE_ROOT}/paddle/fluid/train/demo
./run.sh ${PADDLE_ROOT} ${WITH_MKL:-ON} ./run.sh ${PADDLE_ROOT} ${WITH_MKL:-ON}
EXIT_CODE=$?
fluid_train_endTime_s=`date +%s` fluid_train_endTime_s=`date +%s`
echo "test_fluid_lib_train Total Time: $[ $fluid_train_endTime_s - $fluid_train_startTime_s ]s" echo "test_fluid_lib_train Total Time: $[ $fluid_train_endTime_s - $fluid_train_startTime_s ]s"
./clean.sh ./clean.sh
if [[ "$EXIT_CODE" != "0" ]]; then
exit 8;
fi
} }
function build_document_preview() { function build_document_preview() {
...@@ -1387,7 +1395,7 @@ function main() { ...@@ -1387,7 +1395,7 @@ function main() {
test_inference) test_inference)
gen_fluid_lib ${parallel_number} gen_fluid_lib ${parallel_number}
test_fluid_lib test_fluid_lib
test_fluid_lib_train #test_fluid_lib_train
;; ;;
test_train) test_train)
gen_fluid_lib ${parallel_number} gen_fluid_lib ${parallel_number}
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册