update gpt scripts, test=document_fix

liu-nlper · Jan 7, 2022 · 9e74b50 · 9e74b50
1 parent a7d8fc1
commit 9e74b50
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/benchmark/run_all.sh b/tests/benchmark/run_all.sh
@@ -51,6 +51,6 @@ do
             fi
             CUDA_VISIBLE_DEVICES=0 bash tests/benchmark/run_benchmark.sh sp ${bs_item} ${fp_item}  200  ${model_item} ${mode_item} ${profile} | tee ${log_path}/${model_item}_${mode_item}_bs${bs_item}_${fp_item}_1gpus 2>&1
             sleep 10
-            CUDA_VISIBLE_DEVICES=0 bash tests/benchmark/run_benchmark.sh mp ${bs_item} ${fp_item}  200  ${model_item} ${mode_item} ${profile} | tee ${log_path}/${model_item}_${mode_item}_bs${bs_item}_${fp_item}_8gpus8p 2>&1
+            CUDA_VISIBLE_DEVICES=0,1,2,3,4,5,6,7 bash tests/benchmark/run_benchmark.sh mp ${bs_item} ${fp_item}  200  ${model_item} ${mode_item} ${profile} | tee ${log_path}/${model_item}_${mode_item}_bs${bs_item}_${fp_item}_8gpus8p 2>&1
 done
 done