diff --git a/tests/test_tipc/benchmark_train.sh b/tests/test_tipc/benchmark_train.sh
index 4b7677c72c423d7ce25e9b317b4a29ac919746b9..7f0382ac5d235b0edb297ee28dc828cee2bee913 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