Test llm (#61554)

tianshuo78520a · web-flow · commit 9e18a72bcdf2 · 2024-02-04T17:51:20.000+08:00
* Test llm

* disable test_gradio
diff --git a/paddle/scripts/paddle_build.sh b/paddle/scripts/paddle_build.sh
@@ -3360,21 +3360,10 @@ function distribute_test() {
     export FLAGS_dynamic_static_unified_comm=True
 
     echo "Start LLM Test"
-    # Disable Test: test_gradio
     cd ${work_dir}/PaddleNLP
-    pids=()
-    env CUDA_VISIBLE_DEVICES=0,1 python -m pytest -s -v tests/llm/test_finetune.py &
-    pids+=($!)
-    env CUDA_VISIBLE_DEVICES=2,3 python -m pytest -s -v tests/llm/test_lora.py tests/llm/test_predictor.py &
-    pids+=($!)
-    env CUDA_VISIBLE_DEVICES=4,5 python -m pytest -s -v tests/llm/test_prefix_tuning.py tests/llm/test_pretrain.py &
-    pids+=($!)
-    env CUDA_VISIBLE_DEVICES=6,7 python -m pytest -s -v tests/llm/test_ptq.py tests/llm/testing_utils.py &
-    pids+=($!)
-
-    for pid in "${pids[@]}"; do
-      wait $pid
-    done
+    # Disable Test: test_gradio
+    rm tests/llm/test_gradio.py
+    python -m pytest -s -v tests/llm --timeout=3600
     echo "End LLM Test"
 
     echo "Start auto_parallel Test"
