apps/FastDeploy
1
0
Fork 0
mirror of https://github.com/PaddlePaddle/FastDeploy.git synced 2026-04-23 00:17:25 +08:00
Code Issues Actions 19 Packages Projects Releases Wiki Activity

[CI] Set high-risk OOM tests for sequential execution (#7268)

Browse Source
This commit is contained in:
YuBaoku
2026-04-09 22:22:57 +08:00
committed by GitHub
parent 924690b791
commit ee73623c76
2 changed files with 15 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
scripts/coverage_run.sh
+14
View File
@@ -42,6 +42,20 @@ classify_tests() {
fi
fi
# Rule 5: high-risk OOM tests (treat as multi_gpu for sequential execution)
if [[ "$test_file" == "tests/entrypoints/cli/test_main.py" ||
"$test_file" == "tests/entrypoints/cli/test_serve.py" ||
"$test_file" == "tests/operators/test_group_swiglu_with_masked.py" ||
"$test_file" == "tests/operators/test_hybrid_mtp_ngram.py" ||
"$test_file" == "tests/operators/test_moe_top_k_select.py" ||
"$test_file" == "tests/operators/test_noaux_tc.py" ||
"$test_file" == "tests/output/test_get_save_output_v1.py" ||
"$test_file" == "tests/output/test_process_batch_draft_tokens.py" ||
"$test_file" == "tests/output/test_process_batch_output.py" ]]; then
echo "multi_gpu"
return
fi
# ========== Single-GPU tests (no port required, can run in parallel) ==========
echo "single_gpu"
}
scripts/unittest_requirement.txt
+1
View File
@@ -9,3 +9,4 @@ jsonschema
aistudio_sdk==0.3.5
pandas
use_triton_in_paddle
arctic_inference @ https://paddle-qa.bj.bcebos.com/ernie/arctic_inference-0.1.3-cp310-cp310-linux_x86_64.whl