From ddd40b32111450e57e7e1c084572e4e09757b78b Mon Sep 17 00:00:00 2001 From: Sam Shleifer Date: Tue, 21 Jul 2020 17:01:07 -0400 Subject: [PATCH] [CI] self-scheduled runner tests examples/ (#5927) --- .github/workflows/self-scheduled.yml | 9 +++++++++ 1 file changed, 9 insertions(+) diff --git a/.github/workflows/self-scheduled.yml b/.github/workflows/self-scheduled.yml index 8f5c9831c3..6c658fffc8 100644 --- a/.github/workflows/self-scheduled.yml +++ b/.github/workflows/self-scheduled.yml @@ -47,3 +47,12 @@ jobs: run: | source .env/bin/activate python -m pytest -n 1 --dist=loadfile -s ./tests/ + - name: Run examples tests on GPU + env: + TF_FORCE_GPU_ALLOW_GROWTH: "true" + OMP_NUM_THREADS: 1 + RUN_SLOW: yes + USE_CUDA: yes + run: | + source .env/bin/activate + python -m pytest -n 1 --dist=loadfile -s examples