From cc7de502679b93a5ed825ce7c0ad78b5a5e6a6bf Mon Sep 17 00:00:00 2001
From: huangyuxin <hyxin2014@126.com>
Date: Sat, 9 Oct 2021 09:38:03 +0000
Subject: [PATCH] revise the run_benchmark.sh

---
 tests/benchmark/run_benchmark.sh | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/tests/benchmark/run_benchmark.sh b/tests/benchmark/run_benchmark.sh
index c745aaaf..c03a08f3 100644
--- a/tests/benchmark/run_benchmark.sh
+++ b/tests/benchmark/run_benchmark.sh
@@ -53,7 +53,7 @@ function _train(){
         echo -e "${model_name}, SUCCESS"
         export job_fail_flag=0
     fi
-    kill -9 `ps -ef|grep 'python'|awk '{print $2}'`
+    trap 'for pid in $(jobs -pr); do kill -KILL $pid; done' INT QUIT TERM
 
     if [ $run_mode = "mp" -a -d mylog ]; then
         rm ${log_file}