Skip to content

Commit

Permalink
more logs
Browse files Browse the repository at this point in the history
  • Loading branch information
jerryzh168 committed Jan 11, 2025
1 parent 9c079a4 commit afb1565
Showing 1 changed file with 3 additions and 1 deletion.
4 changes: 3 additions & 1 deletion tutorials/run_all.sh
Original file line number Diff line number Diff line change
Expand Up @@ -10,6 +10,7 @@ find . -type d | while read dir; do
else
find "$dir" -maxdepth 1 -name "*.py" | while read file; do
filename=$(basename "$file")
echo "filename: $filename"
if [ "$filename" = *"tensor_parallel"* ]; then
echo "Running: torchrun --standalone --nnodes=1 --nproc-per-node=1 $file"
torchrun --standalone --nnodes=1 --nproc-per-node=4 "$file"
Expand All @@ -30,7 +31,8 @@ find . -type d | while read dir; do
fi
done

if [ $FAILED -eq 1 ]; then
echo "Failed: $FAILED"
if (( FAILED == 1 )); then
echo "One or more tests failed"
exit 1
else
Expand Down

0 comments on commit afb1565

Please sign in to comment.