diff --git a/tests/test_tipc/benchmark_train.sh b/tests/test_tipc/benchmark_train.sh
index 4b7677c7..7f0382ac 100644
--- a/tests/test_tipc/benchmark_train.sh
+++ b/tests/test_tipc/benchmark_train.sh
@@ -154,6 +154,7 @@ else
     device_num_list=($device_num)
 fi
 
+PYTHON="${python}" bash test_tipc/barrier.sh
 IFS="|"
 for batch_size in ${batch_size_list[*]}; do 
     for precision in ${fp_items_list[*]}; do