[CI] Disable async-assert for base-a tests in rerun-test (#28965)

2026-06-22 15:29:45 -07:00
parent 770d6b2825
commit a32f7111e4
1 changed files with 9 additions and 1 deletions
@@ -145,7 +145,15 @@ jobs:
            echo "Begin ($i/$total): python3 $cmd"
            echo "."
            file_start=$SECONDS
-            (cd test/ && python3 $cmd -f) || exit 1
+            # base-a perf tests (fwd_occupancy) need async assert off; match
+            # _pr-test-stage.yml's base-a carve-out.
+            test_file="${cmd%% *}"
+            test_file="${test_file%%::*}"
+            async_assert=true
+            if grep -qE 'register_cuda_ci\(.*stage[[:space:]]*=[[:space:]]*"base-a"' "test/$test_file" 2>/dev/null; then
+              async_assert=false
+            fi
+            (cd test/ && SGLANG_ENABLE_ASYNC_ASSERT=$async_assert python3 $cmd -f) || exit 1
            elapsed=$(( SECONDS - file_start ))
            echo "."
            echo "End ($i/$total): elapsed=${elapsed}s"