From 908ee7948c155918ffb437d9adfa8b3ea9704af0 Mon Sep 17 00:00:00 2001
From: andyjpaddle <jiangkaitao@baidu.com>
Date: Thu, 2 Jun 2022 08:37:13 +0000
Subject: [PATCH] update mode

---
 test_tipc/{test_serving_cpp.sh => test_serving_infer_cpp.sh} | 5 +++--
 .../{test_serving_python.sh => test_serving_infer_python.sh} | 5 +++--
 2 files changed, 6 insertions(+), 4 deletions(-)
 rename test_tipc/{test_serving_cpp.sh => test_serving_infer_cpp.sh} (99%)
 rename test_tipc/{test_serving_python.sh => test_serving_infer_python.sh} (99%)

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 c9945416..9aafc6a8 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 4affa417..8bdb27bc 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
-- 
GitLab