serving_build.sh 49.4 KB
Newer Older
G
guru4elephant 已提交
1
#!/usr/bin/env bash
M
fix ci  
MRXLT 已提交
2
set -x
3 4 5 6 7 8 9 10 11 12 13 14
function unsetproxy() {
    HTTP_PROXY_TEMP=$http_proxy
    HTTPS_PROXY_TEMP=$https_proxy
    unset http_proxy
    unset https_proxy
}

function setproxy() {
    export http_proxy=$HTTP_PROXY_TEMP
    export https_proxy=$HTTPS_PROXY_TEMP
}

G
guru4elephant 已提交
15 16 17 18 19
function init() {
    source /root/.bashrc
    set -v
    export PYTHONROOT=/usr
    cd Serving
B
barrierye 已提交
20
    export SERVING_WORKDIR=$PWD
M
fix ci  
MRXLT 已提交
21

D
dongdaxiang 已提交
22
    $PYTHONROOT/bin/python -m pip install -r python/requirements.txt
M
fix ci  
MRXLT 已提交
23
    $PYTHONROOT/bin/python -m pip install paddlepaddle
M
fix ci  
MRXLT 已提交
24

B
barriery 已提交
25 26 27
    export GOPATH=$HOME/go
    export PATH=$PATH:$GOPATH/bin

M
fix ci  
MRXLT 已提交
28
    go env -w GO111MODULE=on
M
MRXLT 已提交
29 30
    go env -w GOPROXY=https://goproxy.cn,direct

M
fix ci  
MRXLT 已提交
31 32 33 34
    go get -u github.com/grpc-ecosystem/grpc-gateway/protoc-gen-grpc-gateway@v1.15.2
    go get -u github.com/grpc-ecosystem/grpc-gateway/protoc-gen-swagger@v1.15.2
    go get -u github.com/golang/protobuf/protoc-gen-go@v1.4.3
    go get -u google.golang.org/grpc@v1.33.0
G
guru4elephant 已提交
35 36 37 38 39 40 41 42 43
}

function check_cmd() {
    eval $@
    if [ $? -ne 0 ]; then
        exit 1
    fi
}

44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64
function rerun() {
    if [ $# -ne 2 ]; then
        echo "usage: rerun command rerun-times"
        exit 1
    fi
    local command=$1
    local times=$2
    for((i=1;i<=${times};i++))
    do
        if [ ${i} != 1 ]; then
            echo "${i}-th run command: ${command}..."
        fi
        eval $command
        if [ $? -eq 0 ]; then
            return 0
        fi
        echo "${i}-th run(command: ${command}) failed."
    done
    exit 1
}

B
barrierye 已提交
65 66 67 68 69 70 71 72 73 74 75
function build_app() {
    local TYPE=$1
    local DIRNAME=build-app-$TYPE
    mkdir $DIRNAME # pwd: /Serving
    cd $DIRNAME # pwd: /Serving/build-app-$TYPE
    case $TYPE in
        CPU|GPU)
            cmake -DPYTHON_INCLUDE_DIR=$PYTHONROOT/include/python2.7/ \
                  -DPYTHON_LIBRARIES=$PYTHONROOT/lib/libpython2.7.so \
                  -DPYTHON_EXECUTABLE=$PYTHONROOT/bin/python \
                  -DAPP=ON ..
M
MRXLT 已提交
76
            rerun "make -j10 >/dev/null" 3 # due to some network reasons, compilation may fail
B
barrierye 已提交
77 78 79 80 81 82 83 84 85 86 87
            pip install -U python/dist/paddle_serving_app* >/dev/null
            ;;
        *)
            echo "error type"
            exit 1
            ;;
    esac
    echo "build app $TYPE part finished as expected."
    cd .. # pwd: /Serving
}

G
guru4elephant 已提交
88
function build_client() {
89
    local TYPE=$1
G
guru4elephant 已提交
90
    local DIRNAME=build-client-$TYPE
91 92
    mkdir $DIRNAME # pwd: /Serving
    cd $DIRNAME # pwd: /Serving/build-client-$TYPE
G
guru4elephant 已提交
93 94 95 96 97
    case $TYPE in
        CPU|GPU)
            cmake -DPYTHON_INCLUDE_DIR=$PYTHONROOT/include/python2.7/ \
                  -DPYTHON_LIBRARIES=$PYTHONROOT/lib64/libpython2.7.so \
                  -DPYTHON_EXECUTABLE=$PYTHONROOT/bin/python \
98
                  -DCLIENT=ON ..
M
MRXLT 已提交
99
            rerun "make -j10 >/dev/null" 3 # due to some network reasons, compilation may fail
Y
Your Name 已提交
100
            pip install -U python/dist/paddle_serving_client* >/dev/null
G
guru4elephant 已提交
101 102 103 104 105 106 107
            ;;
        *)
            echo "error type"
            exit 1
            ;;
    esac
    echo "build client $TYPE part finished as expected."
Y
Your Name 已提交
108
    cd .. # pwd: /Serving
109
    # rm -rf $DIRNAME
G
guru4elephant 已提交
110 111 112
}

function build_server() {
113
    local TYPE=$1
G
guru4elephant 已提交
114
    local DIRNAME=build-server-$TYPE
115 116
    mkdir $DIRNAME # pwd: /Serving
    cd $DIRNAME # pwd: /Serving/build-server-$TYPE
G
guru4elephant 已提交
117 118 119 120 121
    case $TYPE in
        CPU)
            cmake -DPYTHON_INCLUDE_DIR=$PYTHONROOT/include/python2.7/ \
                  -DPYTHON_LIBRARIES=$PYTHONROOT/lib64/libpython2.7.so \
                  -DPYTHON_EXECUTABLE=$PYTHONROOT/bin/python \
G
guru4elephant 已提交
122
                  -DSERVER=ON ..
M
MRXLT 已提交
123
            rerun "make -j10 >/dev/null" 3 # due to some network reasons, compilation may fail
124
            check_cmd "make install -j2 >/dev/null"
Y
Your Name 已提交
125
            pip install -U python/dist/paddle_serving_server* >/dev/null
G
guru4elephant 已提交
126 127 128 129 130
            ;;
        GPU)
            cmake -DPYTHON_INCLUDE_DIR=$PYTHONROOT/include/python2.7/ \
                  -DPYTHON_LIBRARIES=$PYTHONROOT/lib64/libpython2.7.so \
                  -DPYTHON_EXECUTABLE=$PYTHONROOT/bin/python \
131
                  -DSERVER=ON \
G
guru4elephant 已提交
132
                  -DWITH_GPU=ON ..
M
MRXLT 已提交
133
            rerun "make -j10 >/dev/null" 3 # due to some network reasons, compilation may fail
134
            check_cmd "make install -j2 >/dev/null"
Y
Your Name 已提交
135
            pip install -U python/dist/paddle_serving_server* >/dev/null
G
guru4elephant 已提交
136 137 138 139 140 141 142
            ;;
        *)
            echo "error type"
            exit 1
            ;;
    esac
    echo "build server $TYPE part finished as expected."
Y
Your Name 已提交
143
    cd .. # pwd: /Serving
Y
Your Name 已提交
144
    # rm -rf $DIRNAME    for export SERVING_BIN
G
guru4elephant 已提交
145 146
}

147
function kill_server_process() {
Y
Your Name 已提交
148
    ps -ef | grep "serving" | grep -v serving_build | grep -v grep | awk '{print $2}' | xargs kill
B
barrierye 已提交
149
    sleep 1
150 151
}

B
barrierye 已提交
152
function kill_process_by_port() {
B
barrierye 已提交
153
    if [ $# != 1 ]; then
B
barrierye 已提交
154
        echo "usage: kill_process_by_port <PID>"
B
barrierye 已提交
155 156 157 158
        exit 1
    fi
    local PID=$1
    lsof -i:$PID | awk 'NR == 1 {next} {print $2}' | xargs kill
159 160
}

G
guru4elephant 已提交
161
function python_test_fit_a_line() {
Y
Your Name 已提交
162 163
    # pwd: /Serving/python/examples
    cd fit_a_line # pwd: /Serving/python/examples/fit_a_line
G
guru4elephant 已提交
164 165
    sh get_data.sh
    local TYPE=$1
B
barrierye 已提交
166
    export SERVING_BIN=${SERVING_WORKDIR}/build-server-${TYPE}/core/general-server/serving
G
guru4elephant 已提交
167 168 169
    case $TYPE in
        CPU)
            # test rpc
170 171
            check_cmd "python -m paddle_serving_server.serve --model uci_housing_model --port 9393 --thread 4 > /dev/null &"
            sleep 5 # wait for the server to start
G
guru4elephant 已提交
172
            check_cmd "python test_client.py uci_housing_client/serving_client_conf.prototxt > /dev/null"
173
            kill_server_process
B
barrierye 已提交
174

G
guru4elephant 已提交
175
            # test web
176
            unsetproxy # maybe the proxy is used on iPipe, which makes web-test failed.
177 178
            check_cmd "python -m paddle_serving_server.serve --model uci_housing_model --name uci --port 9393 --thread 4 --name uci > /dev/null &"
            sleep 5 # wait for the server to start
M
MRXLT 已提交
179
            check_cmd "curl -H \"Content-Type:application/json\" -X POST -d '{\"feed\":[{\"x\": [0.0137, -0.1136, 0.2553, -0.0692, 0.0582, -0.0727, -0.1583, -0.0584, 0.6283, 0.4919, 0.1856, 0.0795, -0.0332]}], \"fetch\":[\"price\"]}' http://127.0.0.1:9393/uci/prediction"
180
            # check http code
M
fix ci  
MRXLT 已提交
181
            http_code=`curl -H "Content-Type:application/json" -X POST -d '{"feed":[{"x": [0.0137, -0.1136, 0.2553, -0.0692, 0.0582, -0.0727, -0.1583, -0.0584, 0.6283, 0.4919, 0.1856, 0.0795, -0.0332]}], "fetch":["price"]}' -s -w "%{http_code}" -o /dev/null http://127.0.0.1:9393/uci/prediction`
Y
Your Name 已提交
182 183 184 185
            if [ ${http_code} -ne 200 ]; then
                echo "HTTP status code -ne 200"
                exit 1
            fi
B
barrierye 已提交
186 187 188 189 190 191 192 193 194 195
            # test web batch
            check_cmd "curl -H \"Content-Type:application/json\" -X POST -d '{\"feed\":[{\"x\": [0.0137, -0.1136, 0.2553, -0.0692, 0.0582, -0.0727, -0.1583, -0.0584, 0.6283, 0.4919, 0.1856, 0.0795, -0.0332]}, {\"x\": [0.0137, -0.1136, 0.2553, -0.0692, 0.0582, -0.0727, -0.1583, -0.0584, 0.6283, 0.4919, 0.1856, 0.0795, -0.0332]}], \"fetch\":[\"price\"]}' http://127.0.0.1:9393/uci/prediction"
            # check http code
            http_code=`curl -H "Content-Type:application/json" -X POST -d '{"feed":[{"x": [0.0137, -0.1136, 0.2553, -0.0692, 0.0582, -0.0727, -0.1583, -0.0584, 0.6283, 0.4919, 0.1856, 0.0795, -0.0332]}, {"x": [0.0137, -0.1136, 0.2553, -0.0692, 0.0582, -0.0727, -0.1583, -0.0584, 0.6283, 0.4919, 0.1856, 0.0795, -0.0332]}], "fetch":["price"]}' -s -w "%{http_code}" -o /dev/null http://127.0.0.1:9393/uci/prediction`
            if [ ${http_code} -ne 200 ]; then
                echo "HTTP status code -ne 200"
                exit 1
            fi
            setproxy # recover proxy state
            kill_server_process
G
guru4elephant 已提交
196 197
            ;;
        GPU)
B
fix bug  
barrierye 已提交
198
            export CUDA_VISIBLE_DEVICES=0
199 200 201 202 203
            # test rpc
            check_cmd "python -m paddle_serving_server_gpu.serve --model uci_housing_model --port 9393 --thread 4 --gpu_ids 0 > /dev/null &"
            sleep 5 # wait for the server to start
            check_cmd "python test_client.py uci_housing_client/serving_client_conf.prototxt > /dev/null"
            kill_server_process
M
MRXLT 已提交
204

205
            # test web
J
Jiawei Wang 已提交
206
            #unsetproxy # maybe the proxy is used on iPipe, which makes web-test failed.
D
dyning 已提交
207 208 209
            #check_cmd "python -m paddle_serving_server_gpu.serve --model uci_housing_model --port 9393 --thread 2 --gpu_ids 0 --name uci > /dev/null &"
            #sleep 5 # wait for the server to start
            #check_cmd "curl -H \"Content-Type:application/json\" -X POST -d '{\"feed\":[{\"x\": [0.0137, -0.1136, 0.2553, -0.0692, 0.0582, -0.0727, -0.1583, -0.0584, 0.6283, 0.4919, 0.1856, 0.0795, -0.0332]}], \"fetch\":[\"price\"]}' http://127.0.0.1:9393/uci/prediction"
210
            # check http code
D
dyning 已提交
211 212 213 214 215
            #http_code=`curl -H "Content-Type:application/json" -X POST -d '{"feed":[{"x": [0.0137, -0.1136, 0.2553, -0.0692, 0.0582, -0.0727, -0.1583, -0.0584, 0.6283, 0.4919, 0.1856, 0.0795, -0.0332]}], "fetch":["price"]}' -s -w "%{http_code}" -o /dev/null http://127.0.0.1:9393/uci/prediction`
            #if [ ${http_code} -ne 200 ]; then
            #    echo "HTTP status code -ne 200"
            #    exit 1
            #fi
B
barrierye 已提交
216
            # test web batch
D
dyning 已提交
217
            #check_cmd "curl -H \"Content-Type:application/json\" -X POST -d '{\"feed\":[{\"x\": [0.0137, -0.1136, 0.2553, -0.0692, 0.0582, -0.0727, -0.1583, -0.0584, 0.6283, 0.4919, 0.1856, 0.0795, -0.0332]}, {\"x\": [0.0137, -0.1136, 0.2553, -0.0692, 0.0582, -0.0727, -0.1583, -0.0584, 0.6283, 0.4919, 0.1856, 0.0795, -0.0332]}], \"fetch\":[\"price\"]}' http://127.0.0.1:9393/uci/prediction"
B
barrierye 已提交
218
            # check http code
D
dyning 已提交
219 220 221 222 223
            #http_code=`curl -H "Content-Type:application/json" -X POST -d '{"feed":[{"x": [0.0137, -0.1136, 0.2553, -0.0692, 0.0582, -0.0727, -0.1583, -0.0584, 0.6283, 0.4919, 0.1856, 0.0795, -0.0332]}, {"x": [0.0137, -0.1136, 0.2553, -0.0692, 0.0582, -0.0727, -0.1583, -0.0584, 0.6283, 0.4919, 0.1856, 0.0795, -0.0332]}], "fetch":["price"]}' -s -w "%{http_code}" -o /dev/null http://127.0.0.1:9393/uci/prediction`
            #if [ ${http_code} -ne 200 ]; then
            #    echo "HTTP status code -ne 200"
            #    exit 1
            #fi
J
Jiawei Wang 已提交
224
            #setproxy # recover proxy state
D
dyning 已提交
225
            #kill_server_process
G
guru4elephant 已提交
226 227 228 229 230 231 232 233
            ;;
        *)
            echo "error type"
            exit 1
            ;;
    esac
    echo "test fit_a_line $TYPE part finished as expected."
    rm -rf image kvdb log uci_housing* work*
B
barrierye 已提交
234
    unset SERVING_BIN
Y
Your Name 已提交
235
    cd .. # pwd: /Serving/python/examples
G
guru4elephant 已提交
236 237
}

W
wangjiawei04 已提交
238
function python_run_criteo_ctr_with_cube() {
Y
merge  
Your Name 已提交
239
    # pwd: /Serving/python/examples
W
wangjiawei04 已提交
240
    local TYPE=$1
241
    yum install -y bc >/dev/null
Y
merge  
Your Name 已提交
242
    cd criteo_ctr_with_cube # pwd: /Serving/python/examples/criteo_ctr_with_cube
M
MRXLT 已提交
243
    export SERVING_BIN=${SERVING_WORKDIR}/build-server-${TYPE}/core/general-server/serving
B
barrierye 已提交
244 245 246 247 248 249 250 251
    case $TYPE in
        CPU)
            check_cmd "wget https://paddle-serving.bj.bcebos.com/unittest/ctr_cube_unittest.tar.gz"
            check_cmd "tar xf ctr_cube_unittest.tar.gz"
            check_cmd "mv models/ctr_client_conf ./"
            check_cmd "mv models/ctr_serving_model_kv ./"
            check_cmd "mv models/data ./cube/"
            check_cmd "mv models/ut_data ./"
M
MRXLT 已提交
252
            cp ../../../build-server-$TYPE/output/bin/cube* ./cube/
B
barrierye 已提交
253 254
            sh cube_prepare.sh &
            python test_server.py ctr_serving_model_kv &
M
MRXLT 已提交
255
            sleep 5
B
barrierye 已提交
256
            check_cmd "python test_client.py ctr_client_conf/serving_client_conf.prototxt ./ut_data >score"
W
wangjiawei04 已提交
257
            tail -n 2 score | awk 'NR==1'
B
barrierye 已提交
258
            AUC=$(tail -n 2  score | awk 'NR==1')
259
            VAR2="0.67" #TODO: temporarily relax the threshold to 0.67
B
barrierye 已提交
260 261
            RES=$( echo "$AUC>$VAR2" | bc )
            if [[ $RES -eq 0 ]]; then
J
Jiawei Wang 已提交
262
                echo "error with criteo_ctr_with_cube inference auc test, auc should > 0.67"
B
barrierye 已提交
263 264 265
                exit 1
            fi
            echo "criteo_ctr_with_cube inference auc test success"
B
barrierye 已提交
266
            kill_server_process
B
barrierye 已提交
267
            ps -ef | grep "cube" | grep -v grep | awk '{print $2}' | xargs kill
B
barrierye 已提交
268
            sleep 1
B
barrierye 已提交
269 270
            ;;
        GPU)
W
wangjiawei04 已提交
271 272 273 274 275 276 277 278 279
            check_cmd "wget https://paddle-serving.bj.bcebos.com/unittest/ctr_cube_unittest.tar.gz"
            check_cmd "tar xf ctr_cube_unittest.tar.gz"
            check_cmd "mv models/ctr_client_conf ./"
            check_cmd "mv models/ctr_serving_model_kv ./"
            check_cmd "mv models/data ./cube/"
            check_cmd "mv models/ut_data ./"
            cp ../../../build-server-$TYPE/output/bin/cube* ./cube/
            sh cube_prepare.sh &
            python test_server_gpu.py ctr_serving_model_kv &
M
MRXLT 已提交
280
            sleep 5
B
barrierye 已提交
281
            # for warm up
B
barrierye 已提交
282
            python test_client.py ctr_client_conf/serving_client_conf.prototxt ./ut_data > /dev/null || true
W
wangjiawei04 已提交
283 284 285
            check_cmd "python test_client.py ctr_client_conf/serving_client_conf.prototxt ./ut_data >score"
            tail -n 2 score | awk 'NR==1'
            AUC=$(tail -n 2  score | awk 'NR==1')
J
Jiawei Wang 已提交
286
            VAR2="0.67" #TODO: temporarily relax the threshold to 0.67
W
wangjiawei04 已提交
287 288
            RES=$( echo "$AUC>$VAR2" | bc )
            if [[ $RES -eq 0 ]]; then
J
Jiawei Wang 已提交
289
                echo "error with criteo_ctr_with_cube inference auc test, auc should > 0.67"
W
wangjiawei04 已提交
290 291 292
                exit 1
            fi
            echo "criteo_ctr_with_cube inference auc test success"
B
barrierye 已提交
293
            kill_server_process
W
wangjiawei04 已提交
294
            ps -ef | grep "cube" | grep -v grep | awk '{print $2}' | xargs kill
B
barrierye 已提交
295
            sleep 1
B
barrierye 已提交
296 297 298 299 300 301
            ;;
        *)
            echo "error type"
            exit 1
            ;;
    esac
M
MRXLT 已提交
302
    unset SERVING_BIN
B
barrierye 已提交
303
    echo "test criteo_ctr_with_cube $TYPE part finished as expected."
Y
merge  
Your Name 已提交
304
    cd .. # pwd: /Serving/python/examples
W
wangjiawei04 已提交
305 306
}

W
wangjiawei04 已提交
307 308 309
function python_test_bert() {
    # pwd: /Serving/python/examples
    local TYPE=$1
310
    export SERVING_BIN=${SERVING_WORKDIR}/build-server-${TYPE}/core/general-server/serving
W
wangjiawei04 已提交
311 312 313
    cd bert # pwd: /Serving/python/examples/bert
    case $TYPE in
        CPU)
314 315 316 317
            # Because download from paddlehub may timeout,
            # download the model from bos(max_seq_len=128).
            wget https://paddle-serving.bj.bcebos.com/paddle_hub_models/text/SemanticModel/bert_chinese_L-12_H-768_A-12.tar.gz
            tar -xzf bert_chinese_L-12_H-768_A-12.tar.gz
W
wangjiawei04 已提交
318
            sh get_data.sh
319
            check_cmd "python -m paddle_serving_server.serve --model bert_chinese_L-12_H-768_A-12_model --port 9292 &"
320
            sleep 5
321
            check_cmd "head -n 10 data-c.txt | python bert_client.py --model bert_chinese_L-12_H-768_A-12_client/serving_client_conf.prototxt"
J
Jiawei Wang 已提交
322
            kill_server_process
M
MRXLT 已提交
323
            echo "bert RPC inference pass"
W
wangjiawei04 已提交
324 325
            ;;
        GPU)
B
fix bug  
barrierye 已提交
326
            export CUDA_VISIBLE_DEVICES=0
327 328 329 330
            # Because download from paddlehub may timeout,
            # download the model from bos(max_seq_len=128).
            wget https://paddle-serving.bj.bcebos.com/paddle_hub_models/text/SemanticModel/bert_chinese_L-12_H-768_A-12.tar.gz
            tar -xzf bert_chinese_L-12_H-768_A-12.tar.gz
W
wangjiawei04 已提交
331
            sh get_data.sh
332
            check_cmd "python -m paddle_serving_server_gpu.serve --model bert_chinese_L-12_H-768_A-12_model --port 9292 --gpu_ids 0 &"
333
            sleep 5
334
            check_cmd "head -n 10 data-c.txt | python bert_client.py --model bert_chinese_L-12_H-768_A-12_client/serving_client_conf.prototxt"
J
Jiawei Wang 已提交
335
            kill_server_process
W
wangjiawei04 已提交
336 337 338
            echo "bert RPC inference pass"
            ;;
        *)
B
barrierye 已提交
339 340 341
            echo "error type"
            exit 1
            ;;
W
wangjiawei04 已提交
342 343
    esac
    echo "test bert $TYPE finished as expected."
344
    unset SERVING_BIN
W
wangjiawei04 已提交
345 346 347
    cd ..
}

M
MRXLT 已提交
348
function python_test_multi_fetch() {
M
MRXLT 已提交
349 350 351 352 353 354 355 356 357
    # pwd: /Serving/python/examples
    local TYPT=$1
    export SERVING_BIN=${SERVING_WORKDIR}/build-server-${TYPE}/core/general-server/serving
    cd bert # pwd: /Serving/python/examples/bert
    case $TYPE in
        CPU)
            #download model (max_seq_len=32)
            wget https://paddle-serving.bj.bcebos.com/bert_example/bert_multi_fetch.tar.gz
            tar -xzvf bert_multi_fetch.tar.gz
M
fix ci  
MRXLT 已提交
358
            check_cmd "python -m paddle_serving_server.serve --model bert_seq32_model --port 9292 &"
M
MRXLT 已提交
359
            sleep 5
M
MRXLT 已提交
360
            check_cmd "head -n 8 data-c.txt | python test_multi_fetch_client.py"
M
MRXLT 已提交
361 362 363 364 365 366 367
            kill_server_process
            echo "bert mutli fetch RPC inference pass"
            ;;
        GPU)
            #download model (max_seq_len=32)
            wget https://paddle-serving.bj.bcebos.com/bert_example/bert_multi_fetch.tar.gz
            tar -xzvf bert_multi_fetch.tar.gz
M
fix ci  
MRXLT 已提交
368
            check_cmd "python -m paddle_serving_server_gpu.serve --model bert_seq32_model --port 9292 --gpu_ids 0 &"
M
MRXLT 已提交
369
            sleep 5
M
MRXLT 已提交
370
            check_cmd "head -n 8 data-c.txt | python test_multi_fetch_client.py"
M
MRXLT 已提交
371 372 373 374 375 376 377 378 379 380 381 382 383 384
            kill_server_process
            echo "bert mutli fetch RPC inference pass"
            ;;
        *)
            echo "error type"
            exit 1
            ;;
    esac
    echo "test multi fetch $TYPE finished as expected."
    unset SERVING_BIN
    cd ..
}

function python_test_multi_process(){
M
MRXLT 已提交
385 386 387
    # pwd: /Serving/python/examples
    local TYPT=$1
    export SERVING_BIN=${SERVING_WORKDIR}/build-server-${TYPE}/core/general-server/serving
M
fix ci  
MRXLT 已提交
388
    cd fit_a_line # pwd: /Serving/python/examples/fit_a_line
M
fix bug  
MRXLT 已提交
389
    sh get_data.sh
M
MRXLT 已提交
390 391
    case $TYPE in
        CPU)
B
barrierye 已提交
392 393
            check_cmd "python -m paddle_serving_server.serve --model uci_housing_model --port 9292 --workdir test9292 &"
            check_cmd "python -m paddle_serving_server.serve --model uci_housing_model --port 9293 --workdir test9293 &"
M
MRXLT 已提交
394 395 396 397 398 399
            sleep 5
            check_cmd "python test_multi_process_client.py"
            kill_server_process
            echo "bert mutli rpc RPC inference pass"
            ;;
        GPU)
B
barrierye 已提交
400
            rm -rf ./image #TODO: The following code tried to create this folder, but no corresponding code was found
B
barrierye 已提交
401 402
            check_cmd "python -m paddle_serving_server_gpu.serve --model uci_housing_model --port 9292 --workdir test9292 --gpu_ids 0 &"
            check_cmd "python -m paddle_serving_server_gpu.serve --model uci_housing_model --port 9293 --workdir test9293 --gpu_ids 0 &"
M
MRXLT 已提交
403 404 405 406 407 408 409 410 411 412 413 414 415
            sleep 5
            check_cmd "python test_multi_process_client.py"
            kill_server_process
            echo "bert mutli process RPC inference pass"
            ;;
        *)
            echo "error type"
            exit 1
            ;;
    esac
    echo "test multi process $TYPE finished as expected."
    unset SERVING_BIN
    cd ..
M
MRXLT 已提交
416 417
}

W
wangjiawei04 已提交
418 419 420
function python_test_imdb() {
    # pwd: /Serving/python/examples
    local TYPE=$1
421 422
    export SERVING_BIN=${SERVING_WORKDIR}/build-server-${TYPE}/core/general-server/serving
    cd imdb # pwd: /Serving/python/examples/imdb
W
wangjiawei04 已提交
423 424 425
    case $TYPE in
        CPU)
            sh get_data.sh
J
Jiawei Wang 已提交
426
            check_cmd "python -m paddle_serving_server.serve --model imdb_cnn_model/ --port 9292 &"
B
barrierye 已提交
427
            sleep 5
W
wangjiawei04 已提交
428
            check_cmd "head test_data/part-0 | python test_client.py imdb_cnn_client_conf/serving_client_conf.prototxt imdb.vocab"
B
barrierye 已提交
429
            # test batch predict
M
MRXLT 已提交
430
            check_cmd "python benchmark.py --thread 4 --batch_size 8 --model imdb_bow_client_conf/serving_client_conf.prototxt --request rpc --endpoint 127.0.0.1:9292"
W
wangjiawei04 已提交
431
            echo "imdb CPU RPC inference pass"
B
barrierye 已提交
432
            kill_server_process
J
Jiawei Wang 已提交
433 434
            rm -rf work_dir1
            sleep 5
W
wangjiawei04 已提交
435

B
barrierye 已提交
436 437
            unsetproxy # maybe the proxy is used on iPipe, which makes web-test failed.
            check_cmd "python text_classify_service.py imdb_cnn_model/ workdir/ 9292 imdb.vocab &"
438
            sleep 5
M
MRXLT 已提交
439
            check_cmd "curl -H \"Content-Type:application/json\" -X POST -d '{\"feed\":[{\"words\": \"i am very sad | 0\"}], \"fetch\":[\"prediction\"]}' http://127.0.0.1:9292/imdb/prediction"
B
barrierye 已提交
440
            http_code=`curl -H "Content-Type:application/json" -X POST -d '{"feed":[{"words": "i am very sad | 0"}], "fetch":["prediction"]}' -s -w "%{http_code}" -o /dev/null http://127.0.0.1:9292/imdb/prediction`
B
barrierye 已提交
441 442 443 444
            if [ ${http_code} -ne 200 ]; then
                echo "HTTP status code -ne 200"
                exit 1
            fi
B
barrierye 已提交
445
            # test batch predict
M
MRXLT 已提交
446
            check_cmd "python benchmark.py --thread 4 --batch_size 8 --model imdb_bow_client_conf/serving_client_conf.prototxt --request http --endpoint 127.0.0.1:9292"
B
barrierye 已提交
447 448 449
            setproxy # recover proxy state
            kill_server_process
            ps -ef | grep "text_classify_service.py" | grep -v grep | awk '{print $2}' | xargs kill
B
barrierye 已提交
450
            echo "imdb CPU HTTP inference pass"
W
wangjiawei04 已提交
451 452 453 454 455
            ;;
        GPU)
            echo "imdb ignore GPU test"
            ;;
        *)
B
barrierye 已提交
456 457 458
            echo "error type"
            exit 1
            ;;
W
wangjiawei04 已提交
459 460
    esac
    echo "test imdb $TYPE finished as expected."
461 462 463 464 465 466 467 468 469 470 471
    unset SERVING_BIN
    cd ..
}

function python_test_lac() {
    # pwd: /Serving/python/examples
    local TYPE=$1
    export SERVING_BIN=${SERVING_WORKDIR}/build-server-${TYPE}/core/general-server/serving
    cd lac # pwd: /Serving/python/examples/lac
    case $TYPE in
        CPU)
M
fix ci  
MRXLT 已提交
472 473 474
            python -m paddle_serving_app.package --get_model lac
            tar -xzvf lac.tar.gz
            check_cmd "python -m paddle_serving_server.serve --model lac_model/ --port 9292 &"
475
            sleep 5
M
fix ci  
MRXLT 已提交
476
            check_cmd "echo \"我爱北京天安门\" | python lac_client.py lac_client/serving_client_conf.prototxt "
477
            echo "lac CPU RPC inference pass"
B
barrierye 已提交
478
            kill_server_process
479

B
barrierye 已提交
480
            unsetproxy # maybe the proxy is used on iPipe, which makes web-test failed.
M
fix ci  
MRXLT 已提交
481
            check_cmd "python lac_web_service.py lac_model/ lac_workdir 9292 &"
482
            sleep 5
B
barrierye 已提交
483 484
            check_cmd "curl -H \"Content-Type:application/json\" -X POST -d '{\"feed\":[{\"words\": \"我爱北京天安门\"}], \"fetch\":[\"word_seg\"]}' http://127.0.0.1:9292/lac/prediction"
            # check http code
B
barrierye 已提交
485
            http_code=`curl -H "Content-Type:application/json" -X POST -d '{"feed":[{"words": "我爱北京天安门"}], "fetch":["word_seg"]}' -s -w "%{http_code}" -o /dev/null http://127.0.0.1:9292/lac/prediction`
B
barrierye 已提交
486 487 488 489 490 491 492
            if [ ${http_code} -ne 200 ]; then
                echo "HTTP status code -ne 200"
                exit 1
            fi
            # http batch
            check_cmd "curl -H \"Content-Type:application/json\" -X POST -d '{\"feed\":[{\"words\": \"我爱北京天安门\"}, {\"words\": \"我爱北京天安门\"}], \"fetch\":[\"word_seg\"]}' http://127.0.0.1:9292/lac/prediction"
            # check http code
B
barrierye 已提交
493
            http_code=`curl -H "Content-Type:application/json" -X POST -d '{"feed":[{"words": "我爱北京天安门"}, {"words": "我爱北京天安门"}], "fetch":["word_seg"]}' -s -w "%{http_code}" -o /dev/null http://127.0.0.1:9292/lac/prediction`
B
barrierye 已提交
494 495 496 497
            if [ ${http_code} -ne 200 ]; then
                echo "HTTP status code -ne 200"
                exit 1
            fi
B
barrierye 已提交
498
            setproxy # recover proxy state
B
barrierye 已提交
499
            kill_server_process
500
            ps -ef | grep "lac_web_service" | grep -v grep | awk '{print $2}' | xargs kill
B
barrierye 已提交
501
            sleep 1
502 503 504 505 506 507
            echo "lac CPU HTTP inference pass"
            ;;
        GPU)
            echo "lac ignore GPU test"
            ;;
        *)
B
barrierye 已提交
508 509 510
            echo "error type"
            exit 1
            ;;
511 512 513
    esac
    echo "test lac $TYPE finished as expected."
    unset SERVING_BIN
W
wangjiawei04 已提交
514 515 516
    cd ..
}

M
MRXLT 已提交
517 518 519 520 521 522 523 524 525

function python_test_encryption(){
    #pwd: /Serving/python/examples
    cd encryption
    sh get_data.sh
    local TYPE=$1
    export SERVING_BIN=${SERIVNG_WORKDIR}/build-server-${TYPE}/core/general-server/serving
    case $TYPE in
        CPU)
M
fix ci  
MRXLT 已提交
526 527
            #check_cmd "python encrypt.py"
            #sleep 5
M
MRXLT 已提交
528 529
            check_cmd "python -m paddle_serving_server.serve --model encrypt_server/ --port 9300 --use_encryption_model > /dev/null &"
            sleep 5
M
fix ci  
MRXLT 已提交
530
            check_cmd "python test_client.py encrypt_client/serving_client_conf.prototxt"
M
MRXLT 已提交
531 532 533
            kill_server_process
            ;;
        GPU)
M
fix ci  
MRXLT 已提交
534 535
            #check_cmd "python encrypt.py"
            #sleep 5
M
MRXLT 已提交
536 537
            check_cmd "python -m paddle_serving_server_gpu.serve --model encrypt_server/ --port 9300 --use_encryption_model --gpu_ids 0"
            sleep 5
M
fix ci  
MRXLT 已提交
538
            check_cmd "python test_client.py encrypt_client/serving_client_conf.prototxt"
M
MRXLT 已提交
539 540 541 542 543 544 545 546 547 548 549 550
            kill_servere_process
            ;;
        *)
            echo "error type"
            exit 1
            ;;
    esac
    echo "encryption $TYPE test finished as expected"
    setproxy
    unset SERVING_BIN
    cd ..
}
B
barrierye 已提交
551 552 553 554 555 556 557
function java_run_test() {
    # pwd: /Serving
    local TYPE=$1
    export SERVING_BIN=${SERVING_WORKDIR}/build-server-${TYPE}/core/general-server/serving
    unsetproxy
    case $TYPE in
        CPU)
B
barrierye 已提交
558 559 560 561 562 563 564 565
            # compile java sdk
            cd java # pwd: /Serving/java
            mvn compile > /dev/null
            mvn install > /dev/null
            # compile java sdk example
            cd examples # pwd: /Serving/java/examples
            mvn compile > /dev/null
            mvn install > /dev/null
M
MRXLT 已提交
566

B
barrierye 已提交
567 568 569 570 571 572 573 574 575 576 577 578 579
            # fit_a_line (general, asyn_predict, batch_predict)
            cd ../../python/examples/grpc_impl_example/fit_a_line # pwd: /Serving/python/examples/grpc_impl_example/fit_a_line
            sh get_data.sh
            check_cmd "python -m paddle_serving_server.serve --model uci_housing_model --port 9393 --thread 4 --use_multilang > /dev/null &"
            sleep 5 # wait for the server to start
            cd ../../../java/examples # /Serving/java/examples
            java -cp target/paddle-serving-sdk-java-examples-0.0.1-jar-with-dependencies.jar PaddleServingClientExample fit_a_line
            java -cp target/paddle-serving-sdk-java-examples-0.0.1-jar-with-dependencies.jar PaddleServingClientExample asyn_predict
            java -cp target/paddle-serving-sdk-java-examples-0.0.1-jar-with-dependencies.jar PaddleServingClientExample batch_predict
            kill_server_process

            # imdb (model_ensemble)
            cd ../../python/examples/grpc_impl_example/imdb # pwd: /Serving/python/examples/grpc_impl_example/imdb
B
barrierye 已提交
580
            sh get_data.sh > /dev/null
B
barrierye 已提交
581 582 583 584 585 586 587 588
            check_cmd "python test_multilang_ensemble_server.py > /dev/null &"
            sleep 5 # wait for the server to start
            cd ../../../java/examples # /Serving/java/examples
            java -cp target/paddle-serving-sdk-java-examples-0.0.1-jar-with-dependencies.jar PaddleServingClientExample model_ensemble
            kill_server_process

            # yolov4 (int32)
            cd ../../python/examples/grpc_impl_example/yolov4 # pwd: /Serving/python/examples/grpc_impl_example/yolov4
B
barrierye 已提交
589 590
            python -m paddle_serving_app.package --get_model yolov4 > /dev/null
            tar -xzf yolov4.tar.gz > /dev/null
B
barrierye 已提交
591
            check_cmd "python -m paddle_serving_server.serve --model yolov4_model --port 9393 --use_multilang --mem_optim > /dev/null &"
B
barrierye 已提交
592 593 594
            cd ../../../java/examples # /Serving/java/examples
            java -cp target/paddle-serving-sdk-java-examples-0.0.1-jar-with-dependencies.jar PaddleServingClientExample yolov4 src/main/resources/000000570688.jpg
            kill_server_process
B
barrierye 已提交
595
            cd ../../ # pwd: /Serving
B
barrierye 已提交
596 597 598 599 600 601 602 603 604 605 606 607 608
            ;;
        GPU)
            ;;
        *)
            echo "error type"
            exit 1
            ;;
    esac
    echo "java-sdk $TYPE part finished as expected."
    setproxy
    unset SERVING_BIN
}

B
barrierye 已提交
609 610 611 612 613
function python_test_grpc_impl() {
    # pwd: /Serving/python/examples
    cd grpc_impl_example # pwd: /Serving/python/examples/grpc_impl_example
    local TYPE=$1
    export SERVING_BIN=${SERVING_WORKDIR}/build-server-${TYPE}/core/general-server/serving
B
barrierye 已提交
614
    unsetproxy
B
barrierye 已提交
615 616 617 618 619 620 621 622 623 624 625 626
    case $TYPE in
        CPU)
            # test general case
            cd fit_a_line # pwd: /Serving/python/examples/grpc_impl_example/fit_a_line
            sh get_data.sh

            # one line command start
            check_cmd "python -m paddle_serving_server.serve --model uci_housing_model --port 9393 --thread 4 --use_multilang > /dev/null &"
            sleep 5 # wait for the server to start
            check_cmd "python test_sync_client.py > /dev/null"
            check_cmd "python test_asyn_client.py > /dev/null"
            check_cmd "python test_general_pb_client.py > /dev/null"
B
barrierye 已提交
627
            check_cmd "python test_numpy_input_client.py > /dev/null"
B
barrierye 已提交
628
            check_cmd "python test_batch_client.py > /dev/null"
B
barrierye 已提交
629
            check_cmd "python test_timeout_client.py > /dev/null"
B
barrierye 已提交
630
            kill_server_process
B
barrierye 已提交
631
            kill_process_by_port 9393
B
barrierye 已提交
632

B
barrierye 已提交
633
            check_cmd "python test_server.py uci_housing_model > /dev/null &"
B
barrierye 已提交
634 635 636 637
            sleep 5 # wait for the server to start
            check_cmd "python test_sync_client.py > /dev/null"
            check_cmd "python test_asyn_client.py > /dev/null"
            check_cmd "python test_general_pb_client.py > /dev/null"
B
barrierye 已提交
638
            check_cmd "python test_numpy_input_client.py > /dev/null"
B
barrierye 已提交
639
            check_cmd "python test_batch_client.py > /dev/null"
B
barrierye 已提交
640
            check_cmd "python test_timeout_client.py > /dev/null"
B
barrierye 已提交
641
            kill_server_process
B
barrierye 已提交
642
            kill_process_by_port 9393
B
barrierye 已提交
643

B
barrierye 已提交
644 645 646 647
            cd .. # pwd: /Serving/python/examples/grpc_impl_example

            # test load server config and client config in Server side
            cd criteo_ctr_with_cube # pwd: /Serving/python/examples/grpc_impl_example/criteo_ctr_with_cube
M
MRXLT 已提交
648
<<COMMENT #comment for compile bug, todo fix conflict between grpc-gateway and cube-agent 
B
barrierye 已提交
649
            check_cmd "wget https://paddle-serving.bj.bcebos.com/unittest/ctr_cube_unittest.tar.gz > /dev/null"
B
barrierye 已提交
650 651 652 653 654 655 656 657 658 659 660 661 662 663 664 665 666 667 668
            check_cmd "tar xf ctr_cube_unittest.tar.gz"
            check_cmd "mv models/ctr_client_conf ./"
            check_cmd "mv models/ctr_serving_model_kv ./"
            check_cmd "mv models/data ./cube/"
            check_cmd "mv models/ut_data ./"
            cp ../../../../build-server-$TYPE/output/bin/cube* ./cube/
            sh cube_prepare.sh &
            check_cmd "mkdir work_dir1 && cp cube/conf/cube.conf ./work_dir1/"
            python test_server.py ctr_serving_model_kv ctr_client_conf/serving_client_conf.prototxt &
            sleep 5
            check_cmd "python test_client.py ./ut_data >score"
            tail -n 2 score | awk 'NR==1'
            AUC=$(tail -n 2  score | awk 'NR==1')
            VAR2="0.67" #TODO: temporarily relax the threshold to 0.67
            RES=$( echo "$AUC>$VAR2" | bc )
            if [[ $RES -eq 0 ]]; then
                echo "error with criteo_ctr_with_cube inference auc test, auc should > 0.67"
                exit 1
            fi
M
MRXLT 已提交
669 670
COMMENT

B
barrierye 已提交
671 672
            echo "grpc impl test success"
            kill_server_process
M
MRXLT 已提交
673
            #ps -ef | grep "cube" | grep -v grep | awk '{print $2}' | xargs kill
B
barrierye 已提交
674 675

            cd .. # pwd: /Serving/python/examples/grpc_impl_example
B
barrierye 已提交
676 677 678 679 680 681 682 683 684 685 686 687 688
            ;;
        GPU)
            export CUDA_VISIBLE_DEVICES=0
            # test general case
            cd fit_a_line # pwd: /Serving/python/examples/grpc_impl_example/fit_a_line
            sh get_data.sh

            # one line command start
            check_cmd "python -m paddle_serving_server_gpu.serve --model uci_housing_model --port 9393 --thread 4 --gpu_ids 0 --use_multilang > /dev/null &"
            sleep 5 # wait for the server to start
            check_cmd "python test_sync_client.py > /dev/null"
            check_cmd "python test_asyn_client.py > /dev/null"
            check_cmd "python test_general_pb_client.py > /dev/null"
B
barrierye 已提交
689
            check_cmd "python test_numpy_input_client.py > /dev/null"
B
barrierye 已提交
690
            check_cmd "python test_batch_client.py > /dev/null"
B
barrierye 已提交
691
            check_cmd "python test_timeout_client.py > /dev/null"
B
barrierye 已提交
692
            kill_server_process
B
barrierye 已提交
693
            kill_process_by_port 9393
B
barrierye 已提交
694

B
barrierye 已提交
695
            check_cmd "python test_server_gpu.py uci_housing_model > /dev/null &"
B
barrierye 已提交
696 697 698 699
            sleep 5 # wait for the server to start
            check_cmd "python test_sync_client.py > /dev/null"
            check_cmd "python test_asyn_client.py > /dev/null"
            check_cmd "python test_general_pb_client.py > /dev/null"
B
barrierye 已提交
700
            check_cmd "python test_numpy_input_client.py > /dev/null"
B
barrierye 已提交
701
            check_cmd "python test_batch_client.py > /dev/null"
B
barrierye 已提交
702
            check_cmd "python test_timeout_client.py > /dev/null"
B
barrierye 已提交
703
            kill_server_process
B
barrierye 已提交
704
            kill_process_by_port 9393
B
barrierye 已提交
705
            #ps -ef | grep "test_server_gpu" | grep -v serving_build | grep -v grep | awk '{print $2}' | xargs kill
B
barrierye 已提交
706

B
barrierye 已提交
707 708 709
            cd .. # pwd: /Serving/python/examples/grpc_impl_example

            # test load server config and client config in Server side
M
MRXLT 已提交
710
<<COMMENT #comment for compile bug, todo fix conflict between grpc-gateway and cube-agent 
B
barrierye 已提交
711 712 713 714 715 716 717 718 719 720 721 722 723
            cd criteo_ctr_with_cube # pwd: /Serving/python/examples/grpc_impl_example/criteo_ctr_with_cube

            check_cmd "wget https://paddle-serving.bj.bcebos.com/unittest/ctr_cube_unittest.tar.gz"
            check_cmd "tar xf ctr_cube_unittest.tar.gz"
            check_cmd "mv models/ctr_client_conf ./"
            check_cmd "mv models/ctr_serving_model_kv ./"
            check_cmd "mv models/data ./cube/"
            check_cmd "mv models/ut_data ./"
            cp ../../../../build-server-$TYPE/output/bin/cube* ./cube/
            sh cube_prepare.sh &
            check_cmd "mkdir work_dir1 && cp cube/conf/cube.conf ./work_dir1/"
            python test_server_gpu.py ctr_serving_model_kv ctr_client_conf/serving_client_conf.prototxt &
            sleep 5
B
barrierye 已提交
724 725
            # for warm up
            python test_client.py ./ut_data &> /dev/null || true
B
barrierye 已提交
726 727 728 729 730 731 732 733 734
            check_cmd "python test_client.py ./ut_data >score"
            tail -n 2 score | awk 'NR==1'
            AUC=$(tail -n 2  score | awk 'NR==1')
            VAR2="0.67" #TODO: temporarily relax the threshold to 0.67
            RES=$( echo "$AUC>$VAR2" | bc )
            if [[ $RES -eq 0 ]]; then
                echo "error with criteo_ctr_with_cube inference auc test, auc should > 0.67"
                exit 1
            fi
M
MRXLT 已提交
735
COMMENT
B
barrierye 已提交
736
            echo "grpc impl test success"
M
MRXLT 已提交
737
            kill_server_process
B
barrierye 已提交
738
            ps -ef | grep "test_server_gpu" | grep -v serving_build | grep -v grep | awk '{print $2}' | xargs kill
M
MRXLT 已提交
739
            #ps -ef | grep "cube" | grep -v grep | awk '{print $2}' | xargs kill
B
barrierye 已提交
740
            cd .. # pwd: /Serving/python/examples/grpc_impl_example
M
MRXLT 已提交
741 742 743 744 745 746
            ;;
        *)
            echo "error type"
            exit 1
            ;;
    esac
B
barrierye 已提交
747
    echo "test grpc impl $TYPE part finished as expected."
B
barrierye 已提交
748
    setproxy
M
MRXLT 已提交
749
    unset SERVING_BIN
B
barrierye 已提交
750
    cd .. # pwd: /Serving/python/examples
M
MRXLT 已提交
751 752 753
}


M
MRXLT 已提交
754 755 756 757 758 759 760
function python_test_yolov4(){
    #pwd:/ Serving/python/examples
    local TYPE=$1
    export SERVING_BIN=${SERVING_WORKDIR}/build-server-${TYPE}/core/general-server/serving
    cd yolov4
    case $TYPE in
        CPU)
M
MRXLT 已提交
761
            echo "no implement for cpu type"
M
MRXLT 已提交
762 763 764 765 766 767 768 769 770 771 772 773 774 775 776 777 778 779 780 781
            ;;
        GPU)
            python -m paddle_serving_app.package --get_model yolov4
            tar -xzvf yolov4.tar.gz
            check_cmd "python -m paddle_serving_server_gpu.serve --model yolov4_model/ --port 9393 --gpu_ids 0 &"
            sleep 5
            check_cmd "python test_client.py 000000570688.jpg"
            echo "yolov4 GPU RPC inference pass"
            kill_server_process
            ;;
        *)
            echo "error type"
            exit 1
            ;;
    esac
    echo "test yolov4 $TYPE finished as expected."
    unset SERVING_BIN
    cd ..
}

M
MRXLT 已提交
782 783 784 785 786 787 788 789 790 791 792 793 794 795 796 797 798 799 800 801 802 803 804 805 806 807 808
function python_test_resnet50(){
    #pwd:/ Serving/python/examples
    local TYPE=$1
    export SERVING_BIN=${SERVING_WORKDIR}/build-server-${TYPE}/core/general-server/serving
    cd imagenet
    case $TYPE in
        CPU)
            echo "no implement for cpu type"
            ;;
        GPU)
            sh get_model.sh
            check_cmd"python -m paddle_serving_server_gpu.serve --model ResNet50_vd_model --port 9696 --gpu_ids 0"
            sleep 5
            check_cmd"python resnet50_rpc_client.py ResNet50_vd_client_config/serving_client_conf.prototxt"
            echo "resnet50 GPU RPC inference pass"
            kill_server_process
            ;;
        *)
            echo "error type"
            exit 1
            ;;
    esac
    echo "test resnet $TYPE finished as expected"
    unset SERVING_BIN
    cd ..
}

B
barrierye 已提交
809
function python_test_pipeline(){
B
barriery 已提交
810
    # pwd: /Serving/python/examples
B
barrierye 已提交
811 812 813
    local TYPE=$1
    export SERVING_BIN=${SERVING_WORKDIR}/build-server-${TYPE}/core/general-server/serving
    unsetproxy
B
barriery 已提交
814
    cd pipeline # pwd: /Serving/python/examples/pipeline
B
barrierye 已提交
815 816
    case $TYPE in
        CPU)
B
barriery 已提交
817
            cd imdb_model_ensemble # pwd: /Serving/python/examples/pipeline/imdb_model_ensemble
B
barrierye 已提交
818 819
            # start paddle serving service (brpc)
            sh get_data.sh
B
barrierye 已提交
820 821
            python -m paddle_serving_server.serve --model imdb_cnn_model --port 9292 --workdir test9292 &> cnn.log &
            python -m paddle_serving_server.serve --model imdb_bow_model --port 9393 --workdir test9393 &> bow.log &
B
barrierye 已提交
822
            sleep 5
M
MRXLT 已提交
823

B
barrierye 已提交
824
            # test: thread servicer & thread op
B
barrierye 已提交
825
            cat << EOF > config.yml
B
barriery 已提交
826
rpc_port: 18080
B
bug fix  
barriery 已提交
827
worker_num: 4
B
barrierye 已提交
828 829 830 831 832 833 834
build_dag_each_worker: false
dag:
    is_thread_op: true
    client_type: brpc
    retry: 1
    use_profile: false
EOF
B
barrierye 已提交
835
            python test_pipeline_server.py > /dev/null &
B
barrierye 已提交
836 837 838
            sleep 5
            check_cmd "python test_pipeline_client.py"
            ps -ef | grep "pipeline_server" | grep -v grep | awk '{print $2}' | xargs kill
B
barrierye 已提交
839
            kill_process_by_port 18080
B
barrierye 已提交
840 841

            # test: thread servicer & process op
B
barrierye 已提交
842
            cat << EOF > config.yml
B
barriery 已提交
843
rpc_port: 18080
B
bug fix  
barriery 已提交
844
worker_num: 4
B
barrierye 已提交
845 846 847 848 849 850 851
build_dag_each_worker: false
dag:
    is_thread_op: false
    client_type: brpc
    retry: 1
    use_profile: false
EOF
B
barrierye 已提交
852
            python test_pipeline_server.py > /dev/null &
B
barrierye 已提交
853 854 855
            sleep 5
            check_cmd "python test_pipeline_client.py"
            ps -ef | grep "pipeline_server" | grep -v grep | awk '{print $2}' | xargs kill
B
barrierye 已提交
856
            kill_process_by_port 18080
B
barrierye 已提交
857

B
bug fix  
barriery 已提交
858
            # test: process servicer & process op
B
barrierye 已提交
859
            cat << EOF > config.yml
B
barriery 已提交
860
rpc_port: 18080
B
bug fix  
barriery 已提交
861 862
worker_num: 4
build_dag_each_worker: false
B
barrierye 已提交
863
dag:
B
bug fix  
barriery 已提交
864
    is_thread_op: false
B
barrierye 已提交
865 866 867 868
    client_type: brpc
    retry: 1
    use_profile: false
EOF
B
barrierye 已提交
869
            python test_pipeline_server.py > /dev/null &
B
barrierye 已提交
870 871 872
            sleep 5
            check_cmd "python test_pipeline_client.py"
            ps -ef | grep "pipeline_server" | grep -v grep | awk '{print $2}' | xargs kill
B
barrierye 已提交
873
            kill_process_by_port 18080
B
bug fix  
barriery 已提交
874 875
            
            # test: process servicer & thread op
M
fix ci  
MRXLT 已提交
876 877
            #pip uninstall grpcio -y
            #pip install grpcio --no-binary=grpcio
B
barrierye 已提交
878
            cat << EOF > config.yml
B
barriery 已提交
879
rpc_port: 18080
B
bug fix  
barriery 已提交
880 881
worker_num: 4
build_dag_each_worker: true
B
barrierye 已提交
882 883 884 885 886 887
dag:
    is_thread_op: false
    client_type: brpc
    retry: 1
    use_profile: false
EOF
B
barrierye 已提交
888
            python test_pipeline_server.py > /dev/null &
B
barrierye 已提交
889 890 891
            sleep 5
            check_cmd "python test_pipeline_client.py"
            ps -ef | grep "pipeline_server" | grep -v grep | awk '{print $2}' | xargs kill
B
barrierye 已提交
892
            kill_process_by_port 18080
M
MRXLT 已提交
893

B
barrierye 已提交
894
            kill_server_process
B
barrierye 已提交
895 896
            kill_process_by_port 9292
            kill_process_by_port 9393
B
barrierye 已提交
897 898

            # start paddle serving service (grpc)
B
barrierye 已提交
899 900
            python -m paddle_serving_server.serve --model imdb_cnn_model --port 9292 --use_multilang --workdir test9292 &> cnn.log &
            python -m paddle_serving_server.serve --model imdb_bow_model --port 9393 --use_multilang --workdir test9393 &> bow.log &
B
barrierye 已提交
901
            sleep 5
B
barrierye 已提交
902
            cat << EOF > config.yml
B
barriery 已提交
903
rpc_port: 18080
B
bug fix  
barriery 已提交
904
worker_num: 4
B
barrierye 已提交
905 906 907 908 909 910 911
build_dag_each_worker: false
dag:
    is_thread_op: false
    client_type: grpc
    retry: 1
    use_profile: false
EOF
B
barrierye 已提交
912
            python test_pipeline_server.py > /dev/null &
B
barrierye 已提交
913 914 915
            sleep 5
            check_cmd "python test_pipeline_client.py"
            ps -ef | grep "pipeline_server" | grep -v grep | awk '{print $2}' | xargs kill
B
barrierye 已提交
916
            kill_process_by_port 18080
B
barrierye 已提交
917
            kill_server_process
B
barrierye 已提交
918 919
            kill_process_by_port 9292
            kill_process_by_port 9393
B
barriery 已提交
920
            cd ..
B
fix ci  
barriery 已提交
921

B
fix doc  
barriery 已提交
922
            cd simple_web_service # pwd: /Serving/python/examples/pipeline/simple_web_service
B
barriery 已提交
923 924 925
            sh get_data.sh
            python web_service.py >/dev/null &
            sleep 5
B
barrierye 已提交
926
            curl -X POST -k http://localhost:18080/uci/prediction -d '{"key": ["x"], "value": ["0.0137, -0.1136, 0.2553, -0.0692, 0.0582, -0.0727, -0.1583, -0.0584, 0.6283, 0.4919, 0.1856, 0.0795, -0.0332"]}'
B
barriery 已提交
927
             check http code
B
barrierye 已提交
928
            http_code=`curl -X POST -k -d '{"key":["x"], "value": ["0.0137, -0.1136, 0.2553, -0.0692, 0.0582, -0.0727, -0.1583, -0.0584, 0.6283, 0.4919, 0.1856, 0.0795, -0.0332"]}' -s -w "%{http_code}" -o /dev/null http://localhost:18080/uci/prediction`
B
barriery 已提交
929 930 931 932 933
            if [ ${http_code} -ne 200 ]; then
                echo "HTTP status code -ne 200"
                exit 1
            fi
            ps -ef | grep "web_service" | grep -v grep | awk '{print $2}' | xargs kill
B
barriery 已提交
934
            ps -ef | grep "pipeline" | grep -v grep | awk '{print $2}' | xargs kill
B
barriery 已提交
935 936
            kill_server_process
            cd ..
B
barrierye 已提交
937 938
            ;;
        GPU)
B
fix doc  
barriery 已提交
939
            cd simple_web_service # pwd: /Serving/python/examples/pipeline/simple_web_service
B
barriery 已提交
940 941 942
            sh get_data.sh
            python web_service.py >/dev/null &
            sleep 5
B
barrierye 已提交
943
            curl -X POST -k http://localhost:18080/uci/prediction -d '{"key": ["x"], "value": ["0.0137, -0.1136, 0.2553, -0.0692, 0.0582, -0.0727, -0.1583, -0.0584, 0.6283, 0.4919, 0.1856, 0.0795, -0.0332"]}'
B
barriery 已提交
944
            # check http code
B
barrierye 已提交
945
            http_code=`curl -X POST -k -d '{"key":["x"], "value": ["0.0137, -0.1136, 0.2553, -0.0692, 0.0582, -0.0727, -0.1583, -0.0584, 0.6283, 0.4919, 0.1856, 0.0795, -0.0332"]}' -s -w "%{http_code}" -o /dev/null http://localhost:18080/uci/prediction`
B
barriery 已提交
946 947 948 949 950
            if [ ${http_code} -ne 200 ]; then
                echo "HTTP status code -ne 200"
                exit 1
            fi
            ps -ef | grep "web_service" | grep -v grep | awk '{print $2}' | xargs kill
B
barriery 已提交
951
            ps -ef | grep "pipeline" | grep -v grep | awk '{print $2}' | xargs kill
B
fix ci  
barriery 已提交
952
            kill_server_process
B
barriery 已提交
953
            cd .. # pwd: /Serving/python/examples/pipeline
B
barrierye 已提交
954 955 956 957 958 959
            ;;
        *)
            echo "error type"
            exit 1
            ;;
    esac
B
barriery 已提交
960
    cd ..
B
barrierye 已提交
961 962 963 964
    setproxy
    unset SERVING_BIN
}

M
MRXLT 已提交
965 966 967 968 969 970 971 972 973 974 975 976 977 978 979 980 981 982 983 984
function python_app_api_test(){
    #pwd:/ Serving/python/examples
    #test image reader
    local TYPE=$1
    cd imagenet
    case $TYPE in
        CPU)
            check_cmd "python test_image_reader.py"
            ;;
        GPU)
            echo "no implement for cpu type"
            ;;
        *)
            echo "error type"
            exit 1
            ;;
    esac
    echo "test app api finised as expected"
    cd ..
}
M
MRXLT 已提交
985

G
guru4elephant 已提交
986
function python_run_test() {
987
    # Using the compiled binary
Y
Your Name 已提交
988 989 990
    local TYPE=$1 # pwd: /Serving
    cd python/examples # pwd: /Serving/python/examples
    python_test_fit_a_line $TYPE # pwd: /Serving/python/examples
M
MRXLT 已提交
991
    #python_run_criteo_ctr_with_cube $TYPE # pwd: /Serving/python/examples
W
wangjiawei04 已提交
992
    python_test_bert $TYPE # pwd: /Serving/python/examples
M
MRXLT 已提交
993
    python_test_imdb $TYPE # pwd: /Serving/python/examples
M
MRXLT 已提交
994 995 996
    python_test_lac $TYPE # pwd: /Serving/python/examples
    python_test_multi_process $TYPE # pwd: /Serving/python/examples
    python_test_multi_fetch $TYPE # pwd: /Serving/python/examples
M
fix ci  
MRXLT 已提交
997
    python_test_encryption $TYPE # pwd: /Serving/python/examples
M
MRXLT 已提交
998
    python_test_yolov4 $TYPE # pwd: /Serving/python/examples
B
barrierye 已提交
999
    python_test_grpc_impl $TYPE # pwd: /Serving/python/examples
M
MRXLT 已提交
1000
    python_test_resnet50 $TYPE # pwd: /Serving/python/examples
1001
    #python_test_pipeline $TYPE # pwd: /Serving/python/examples
G
guru4elephant 已提交
1002
    echo "test python $TYPE part finished as expected."
Y
Your Name 已提交
1003
    cd ../.. # pwd: /Serving
G
guru4elephant 已提交
1004 1005
}

B
barrierye 已提交
1006 1007 1008 1009 1010
function monitor_test() {
    local TYPE=$1 # pwd: /Serving
    mkdir _monitor_test && cd _monitor_test # pwd: /Serving/_monitor_test
    case $TYPE in
        CPU):
B
barrierye 已提交
1011 1012
            # The CPU part and GPU part are identical.
            # In order to avoid Travis CI timeout (50 min), the CPU version is not checked
B
barrierye 已提交
1013 1014
            ;;
        GPU):
B
barrierye 已提交
1015 1016 1017 1018 1019 1020 1021 1022 1023 1024 1025 1026 1027 1028 1029 1030 1031 1032 1033 1034 1035 1036 1037 1038 1039 1040 1041 1042 1043 1044 1045 1046 1047 1048 1049 1050 1051 1052 1053 1054 1055 1056 1057 1058 1059 1060 1061 1062 1063 1064 1065 1066 1067 1068 1069 1070 1071 1072 1073 1074 1075 1076 1077 1078 1079 1080 1081 1082 1083 1084 1085 1086 1087 1088 1089 1090 1091 1092 1093 1094 1095 1096 1097 1098 1099 1100 1101 1102 1103 1104 1105 1106 1107 1108 1109 1110 1111 1112 1113 1114 1115 1116 1117 1118 1119 1120 1121 1122 1123 1124 1125 1126
            pip install pyftpdlib
            mkdir remote_path
            mkdir local_path
            cd remote_path # pwd: /Serving/_monitor_test/remote_path
            check_cmd "python -m pyftpdlib -p 8000 &>/dev/null &"
            cd .. # pwd: /Serving/_monitor_test

            # type: ftp
            # remote_path: /
            # remote_model_name: uci_housing.tar.gz
            # local_tmp_path: ___tmp
            # local_path: local_path
            cd remote_path # pwd: /Serving/_monitor_test/remote_path
            wget --no-check-certificate https://paddle-serving.bj.bcebos.com/uci_housing.tar.gz
            touch donefile
            cd ..  # pwd: /Serving/_monitor_test
            mkdir -p local_path/uci_housing_model
            python -m paddle_serving_server_gpu.monitor \
                    --type='ftp' --ftp_host='127.0.0.1' --ftp_port='8000' \
                    --remote_path='/' --remote_model_name='uci_housing.tar.gz' \
                    --remote_donefile_name='donefile' --local_path='local_path' \
                    --local_model_name='uci_housing_model' --local_timestamp_file='fluid_time_file' \
                    --local_tmp_path='___tmp' --unpacked_filename='uci_housing_model' \
                    --interval='1' >/dev/null &
            sleep 10
            if [ ! -f "local_path/uci_housing_model/fluid_time_file" ]; then
                echo "local_path/uci_housing_model/fluid_time_file not exist."
                exit 1
            fi
            ps -ef | grep "monitor" | grep -v grep | awk '{print $2}' | xargs kill
            rm -rf remote_path/*
            rm -rf local_path/*

            # type: ftp
            # remote_path: /tmp_dir
            # remote_model_name: uci_housing_model
            # local_tmp_path: ___tmp
            # local_path: local_path
            mkdir -p remote_path/tmp_dir && cd remote_path/tmp_dir # pwd: /Serving/_monitor_test/remote_path/tmp_dir
            wget --no-check-certificate https://paddle-serving.bj.bcebos.com/uci_housing.tar.gz
            tar -xzf uci_housing.tar.gz
            touch donefile
            cd ../.. # pwd: /Serving/_monitor_test
            mkdir -p local_path/uci_housing_model
            python -m paddle_serving_server_gpu.monitor \
                    --type='ftp' --ftp_host='127.0.0.1' --ftp_port='8000' \
                    --remote_path='/tmp_dir' --remote_model_name='uci_housing_model' \
                    --remote_donefile_name='donefile' --local_path='local_path' \
                    --local_model_name='uci_housing_model' --local_timestamp_file='fluid_time_file' \
                    --local_tmp_path='___tmp' --interval='1' >/dev/null &
            sleep 10
            if [ ! -f "local_path/uci_housing_model/fluid_time_file" ]; then
                echo "local_path/uci_housing_model/fluid_time_file not exist."
                exit 1
            fi
            ps -ef | grep "monitor" | grep -v grep | awk '{print $2}' | xargs kill
            rm -rf remote_path/*
            rm -rf local_path/*

            # type: general
            # remote_path: /
            # remote_model_name: uci_housing.tar.gz
            # local_tmp_path: ___tmp
            # local_path: local_path
            cd remote_path # pwd: /Serving/_monitor_test/remote_path
            wget --no-check-certificate https://paddle-serving.bj.bcebos.com/uci_housing.tar.gz
            touch donefile
            cd ..  # pwd: /Serving/_monitor_test
            mkdir -p local_path/uci_housing_model
            python -m paddle_serving_server_gpu.monitor \
                    --type='general' --general_host='ftp://127.0.0.1:8000' \
                    --remote_path='/' --remote_model_name='uci_housing.tar.gz' \
                    --remote_donefile_name='donefile' --local_path='local_path' \
                    --local_model_name='uci_housing_model' --local_timestamp_file='fluid_time_file' \
                    --local_tmp_path='___tmp' --unpacked_filename='uci_housing_model' \
                    --interval='1' >/dev/null &
            sleep 10
            if [ ! -f "local_path/uci_housing_model/fluid_time_file" ]; then
                echo "local_path/uci_housing_model/fluid_time_file not exist."
                exit 1
            fi
            ps -ef | grep "monitor" | grep -v grep | awk '{print $2}' | xargs kill
            rm -rf remote_path/*
            rm -rf local_path/*

            # type: general
            # remote_path: /tmp_dir
            # remote_model_name: uci_housing_model
            # local_tmp_path: ___tmp
            # local_path: local_path
            mkdir -p remote_path/tmp_dir && cd remote_path/tmp_dir # pwd: /Serving/_monitor_test/remote_path/tmp_dir
            wget --no-check-certificate https://paddle-serving.bj.bcebos.com/uci_housing.tar.gz
            tar -xzf uci_housing.tar.gz
            touch donefile
            cd ../.. # pwd: /Serving/_monitor_test
            mkdir -p local_path/uci_housing_model
            python -m paddle_serving_server_gpu.monitor \
                    --type='general' --general_host='ftp://127.0.0.1:8000' \
                    --remote_path='/tmp_dir' --remote_model_name='uci_housing_model' \
                    --remote_donefile_name='donefile' --local_path='local_path' \
                    --local_model_name='uci_housing_model' --local_timestamp_file='fluid_time_file' \
                    --local_tmp_path='___tmp' --interval='1' >/dev/null &
            sleep 10
            if [ ! -f "local_path/uci_housing_model/fluid_time_file" ]; then
                echo "local_path/uci_housing_model/fluid_time_file not exist."
                exit 1
            fi
            ps -ef | grep "monitor" | grep -v grep | awk '{print $2}' | xargs kill
            rm -rf remote_path/*
            rm -rf local_path/*

            ps -ef | grep "pyftpdlib" | grep -v grep | awk '{print $2}' | xargs kill
B
barrierye 已提交
1127 1128
            ;;
        *)
B
barrierye 已提交
1129 1130 1131
            echo "error type"
            exit 1
            ;;
B
barrierye 已提交
1132 1133 1134 1135 1136 1137
    esac
    cd .. # pwd: /Serving
    rm -rf _monitor_test
    echo "test monitor $TYPE finished as expected."
}

G
guru4elephant 已提交
1138
function main() {
Y
Your Name 已提交
1139 1140 1141 1142
    local TYPE=$1 # pwd: /
    init # pwd: /Serving
    build_client $TYPE # pwd: /Serving
    build_server $TYPE # pwd: /Serving
B
barrierye 已提交
1143
    build_app $TYPE # pwd: /Serving
B
barrierye 已提交
1144
    java_run_test $TYPE # pwd: /Serving
Y
Your Name 已提交
1145
    python_run_test $TYPE # pwd: /Serving
W
wangjiawei04 已提交
1146
    #monitor_test $TYPE # pwd: /Serving
G
guru4elephant 已提交
1147 1148 1149 1150
    echo "serving $TYPE part finished as expected."
}

main $@
B
barrierye 已提交
1151
exit 0