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

Fix inference CI bug (#26080)

* Fix inference bug

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