Skip to content

Commit c299082

Browse files
committed
ci: preserve SAI Slurm logs on failure
1 parent 9bec2a3 commit c299082

1 file changed

Lines changed: 13 additions & 6 deletions

File tree

.github/workflows/sai-gpu-full.yml

Lines changed: 13 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -38,10 +38,13 @@ jobs:
3838
run: |
3939
set -euo pipefail
4040
export CI_SRC=/home/sai-partner/saiuser01/agent/abacus_beta6_build/ci-checkouts/${GITHUB_RUN_ID}
41+
export RESULT_ROOT=/home/sai-partner/saiuser01/agent/abacus_beta6_build/ci-results/${GITHUB_RUN_ID}
4142
rm -rf "$CI_SRC"
42-
mkdir -p "$CI_SRC"
43+
rm -rf "$RESULT_ROOT"
44+
mkdir -p "$CI_SRC" "$RESULT_ROOT"
4345
rsync -a --delete --exclude .git/ "$GITHUB_WORKSPACE/" "$CI_SRC/"
4446
echo "CI_SRC=$CI_SRC" >> "$GITHUB_ENV"
47+
echo "RESULT_ROOT=$RESULT_ROOT" >> "$GITHUB_ENV"
4548
4649
- name: Submit Slurm GPU validation
4750
env:
@@ -60,7 +63,7 @@ jobs:
6063
#SBATCH --ntasks=4
6164
#SBATCH --gpus-per-node=4
6265
#SBATCH --time=01:00:00
63-
#SBATCH --output=__CI_SRC__/sai-gpu-%j.out
66+
#SBATCH --output=__RESULT_ROOT__/sai-gpu-%j.out
6467
6568
set -euo pipefail
6669
: "${CI_SRC:?}"
@@ -118,11 +121,14 @@ jobs:
118121
fi
119122
echo "SAI_GPU_VALIDATION_PASSED mp_version=$MP_VERSION"
120123
SBATCH
121-
sed -i "s|__CI_SRC__|$CI_SRC|g" "$script"
122-
export CI_SRC ABACUS=/home/sai-partner/saiuser01/agent/abacus_beta6_build/install-full-sm70-avx2-mp09/bin/abacus MP_VERSION RUN_CUSOLVERMP
124+
sed -i "s|__RESULT_ROOT__|$RESULT_ROOT|g" "$script"
125+
export CI_SRC RESULT_ROOT ABACUS=/home/sai-partner/saiuser01/agent/abacus_beta6_build/install-full-sm70-avx2-mp09/bin/abacus MP_VERSION RUN_CUSOLVERMP
126+
set +e
123127
sbatch --wait --parsable --export=ALL "$script" | tee "$log"
124-
cat "$CI_SRC"/sai-gpu-*.out >> "$log"
125-
grep -q 'SAI_GPU_VALIDATION_PASSED' "$log"
128+
status=${PIPESTATUS[0]}
129+
set -e
130+
cat "$RESULT_ROOT"/sai-gpu-*.out >> "$log" 2>/dev/null || true
131+
exit "$status"
126132
127133
- name: Upload SAI logs
128134
if: always()
@@ -133,6 +139,7 @@ jobs:
133139
${{ env.CI_SRC }}/**/log.txt
134140
${{ env.CI_SRC }}/**/result.out
135141
${{ env.CI_SRC }}/sai-gpu-${{ github.run_id }}.log
142+
${{ env.RESULT_ROOT }}/sai-gpu-*.out
136143
if-no-files-found: warn
137144

138145
- name: Cleanup shared checkout

0 commit comments

Comments
 (0)