Skip to content

Commit

Permalink
[CI] Add benchmarks to test runs
Browse files Browse the repository at this point in the history
ghstack-source-id: 9e63898503a8f05206cb05de91bac6346615815b
Pull Request resolved: #2410
  • Loading branch information
vmoens committed Sep 2, 2024
1 parent a5ac018 commit 893a3b3
Showing 1 changed file with 5 additions and 0 deletions.
5 changes: 5 additions & 0 deletions .github/unittest/linux/scripts/run_all.sh
Original file line number Diff line number Diff line change
Expand Up @@ -189,9 +189,14 @@ export MKL_THREADING_LAYER=GNU
export CKPT_BACKEND=torch
export MAX_IDLE_COUNT=100
export BATCHED_PIPE_TIMEOUT=60
export TORCHDYNAMO_INLINE_INBUILT_NN_MODULES=1

pytest test/smoke_test.py -v --durations 200
pytest test/smoke_test_deps.py -v --durations 200 -k 'test_gym or test_dm_control_pixels or test_dm_control or test_tb'

# Check that benchmarks run
python -m pytest benchmarks

if [ "${CU_VERSION:-}" != cpu ] ; then
python .github/unittest/helpers/coverage_run_parallel.py -m pytest test \
--instafail --durations 200 -vv --capture no --ignore test/test_rlhf.py \
Expand Down

0 comments on commit 893a3b3

Please sign in to comment.