Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Capture error message during sweep tests #2351

Merged
merged 11 commits into from
Sep 27, 2022
Prev Previous commit
Next Next commit
Revert "Keep running tests even if errors"
This reverts commit dca97c2.
  • Loading branch information
molly-smith committed Sep 23, 2022
commit c5f3b8b4307d21e65afd1f893ce247d07d1b9fec
8 changes: 4 additions & 4 deletions benchmarks/inference/sweep.sh
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
set -x
set -ex

export TRANSFORMERS_CACHE=/tmp/hf-cache

Expand All @@ -8,10 +8,10 @@ branch2=$2
gptneo_models="EleutherAI/gpt-neo-2.7B EleutherAI/gpt-neo-1.3B EleutherAI/gpt-neo-125M"
gpt2_models="gpt2 gpt2-large gpt2-xl"
gptj_models="EleutherAI/gpt-j-6B"
opt_models=""
bloom_models=""
opt_models="facebook/opt-125m facebook/opt-1.3b facebook/opt-2.7b facebook/opt-6.7b facebook/opt-13b"
bloom_models="bigscience/bloom-560m bigscience/bloom-1b7 bigscience/bloom-3b bigscience/bloom-7b1"

for gpus in `echo "1"`; do
for gpus in `echo "1 2 4 8"`; do
for dtype in `echo "fp16 fp32"`; do
for graphs in `echo "true false"`; do
for kernel in `echo "true false"`; do
Expand Down