Skip to content

Commit

Permalink
Update userbenchmark-a100-release.yml
Browse files Browse the repository at this point in the history
  • Loading branch information
juliagmt-google authored Nov 12, 2024
1 parent 63cf7e6 commit 37ac07c
Showing 1 changed file with 6 additions and 0 deletions.
6 changes: 6 additions & 0 deletions .github/workflows/userbenchmark-a100-release.yml
Original file line number Diff line number Diff line change
Expand Up @@ -42,6 +42,12 @@ jobs:
# remove old results
if [ -d benchmark-output ]; then rm -Rf benchmark-output; fi
pushd benchmark
# Install necessary packages <-- Added these lines
pip3 install -e '.[jax_cpu]'
pip3 install -e '.[pytorch_gpu]' -f 'https://download.pytorch.org/whl/cu121'
pip3 install -e '.[full]'
release_version=$(cat userbenchmark/release-test/version.txt)
if [ -d .userbenchmark ]; then rm -Rf .userbenchmark; fi
python run_benchmark.py release-test -c ${release_version}
Expand Down

0 comments on commit 37ac07c

Please sign in to comment.