From 893a3b360d90656e5bbc2161283729b738e95d28 Mon Sep 17 00:00:00 2001 From: Vincent Moens Date: Mon, 2 Sep 2024 16:02:07 +0100 Subject: [PATCH] [CI] Add benchmarks to test runs ghstack-source-id: 9e63898503a8f05206cb05de91bac6346615815b Pull Request resolved: https://github.com/pytorch/rl/pull/2410 --- .github/unittest/linux/scripts/run_all.sh | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/.github/unittest/linux/scripts/run_all.sh b/.github/unittest/linux/scripts/run_all.sh index 3257adf8c63..3421b11f550 100755 --- a/.github/unittest/linux/scripts/run_all.sh +++ b/.github/unittest/linux/scripts/run_all.sh @@ -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 \