From ec54d341eca895edd9aa24285889d638a0759b68 Mon Sep 17 00:00:00 2001 From: xuexixi Date: Fri, 20 Dec 2024 17:22:08 +0800 Subject: [PATCH] fix_recompute_test --- scripts/distribute/ci_case_auto.sh | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/scripts/distribute/ci_case_auto.sh b/scripts/distribute/ci_case_auto.sh index 82b6e8ef8af9..d4f4e2df4a52 100755 --- a/scripts/distribute/ci_case_auto.sh +++ b/scripts/distribute/ci_case_auto.sh @@ -668,6 +668,10 @@ function llama_align_dygraph_dy2st_pir_auto_bs2_bf16_DP2-MP2-PP1-SP() { for to_static in "0" "1"; do for use_recompute in "1" "0"; do + if [ "$to_static" -eq "0" ] && [ "$use_recompute" -eq "1" ]; then + # The test for recompute only runs when `to_static = 1`. + continue + fi rm -rf $case_out_dir rm -rf $case_log_dir python -u -m paddle.distributed.launch \