diff --git a/test_tipc/test_serving_cpp.sh b/test_tipc/test_serving_infer_cpp.sh similarity index 99% rename from test_tipc/test_serving_cpp.sh rename to test_tipc/test_serving_infer_cpp.sh index c9945416b6937e42a2798bde222467948a6d75c9..9aafc6a8278234064b6013b2ce29a0c31df05d52 100644 --- a/test_tipc/test_serving_cpp.sh +++ b/test_tipc/test_serving_infer_cpp.sh @@ -11,6 +11,7 @@ function func_parser_model_config(){ FILENAME=$1 dataline=$(awk 'NR==1, NR==19{print}' $FILENAME) +MODE=$2 # parser params IFS=$'\n' @@ -51,7 +52,7 @@ cpp_client_py=$(func_parser_value "${lines[17]}") image_dir_key=$(func_parser_key "${lines[18]}") image_dir_value=$(func_parser_value "${lines[18]}") -LOG_PATH="$(pwd)/test_tipc/output/${model_name}/cpp_serving" +LOG_PATH="$(pwd)/test_tipc/output/${model_name}/${MODE}/cpp" mkdir -p ${LOG_PATH} status_log="${LOG_PATH}/results_cpp_serving.log" @@ -167,7 +168,7 @@ function func_serving(){ #set cuda device -GPUID=$2 +GPUID=$3 if [ ${#GPUID} -le 0 ];then env="export CUDA_VISIBLE_DEVICES=0" else diff --git a/test_tipc/test_serving_python.sh b/test_tipc/test_serving_infer_python.sh similarity index 99% rename from test_tipc/test_serving_python.sh rename to test_tipc/test_serving_infer_python.sh index 4affa417e10ebf642b7b280dc0b4b373a6b341fe..8bdb27bce7b128f3141c6f1ed774ecc709c71968 100644 --- a/test_tipc/test_serving_python.sh +++ b/test_tipc/test_serving_infer_python.sh @@ -11,6 +11,7 @@ function func_parser_model_config(){ FILENAME=$1 dataline=$(awk 'NR==1, NR==23{print}' $FILENAME) +MODE=$2 # parser params IFS=$'\n' @@ -58,7 +59,7 @@ pipeline_py=$(func_parser_value "${lines[21]}") image_dir_key=$(func_parser_key "${lines[22]}") image_dir_value=$(func_parser_value "${lines[22]}") -LOG_PATH="$(pwd)/test_tipc/output/${model_name}/python_serving" +LOG_PATH="$(pwd)/test_tipc/output/${model_name}/${MODE}/python" mkdir -p ${LOG_PATH} status_log="${LOG_PATH}/results_python_serving.log" @@ -207,7 +208,7 @@ function func_serving(){ #set cuda device -GPUID=$2 +GPUID=$3 if [ ${#GPUID} -le 0 ];then env="export CUDA_VISIBLE_DEVICES=0" else