Skip to content
This repository was archived by the owner on Sep 10, 2025. It is now read-only.

Commit c935680

Browse files
authored
Update more-tests.yml
Display test information to simplify debug
1 parent de5bc74 commit c935680

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

.github/workflows/more-tests.yml

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -19,6 +19,7 @@ jobs:
1919
gpu-arch-version: "12.4"
2020
timeout: 60
2121
script: |
22+
set -xeou pipefail
2223
echo "::group::Print machine info"
2324
uname -a
2425
echo "::endgroup::"
@@ -96,6 +97,7 @@ jobs:
9697
gpu-arch-version: "12.4"
9798
timeout: 60
9899
script: |
100+
set -xeou pipefail
99101
echo "::group::Print machine info"
100102
uname -a
101103
echo "::endgroup::"
@@ -125,6 +127,8 @@ jobs:
125127
# (although the runner environment should not have an opinion what we us in the artifact, and we might suitably abstract that)
126128
for DTYPE in bfloat16 float16 float32; do
127129
for SDPA in 'math' 'flash_attention' 'efficient_attention' 'cudnn_attention'; do
130+
echo "***************************************************************"
131+
echo "*** $DEVICE $DTYPE $SDPA"
128132
###################################################################
129133
# Export DSO and run with Python
130134
python torchchat.py export --output-dso dso.so --checkpoint-path ${MODEL_PATH} --attention-backend ${SDPA} --device ${DEVICE} --dtype ${DTYPE} --temperature 0

0 commit comments

Comments
 (0)