From 4b48be8fcc73cf3951b009b9dfa4a096e0742501 Mon Sep 17 00:00:00 2001
From: Isaac Yang <isaacy@nvidia.com>
Date: Wed, 22 Jan 2020 17:04:00 -0800
Subject: [PATCH] Add script to run unit tests in GPU runner
 (nvcr.io/nvidia/pytorch:19.10-py3)

---
 .gitlab-ci.yml | 12 +++++++++++-
 1 file changed, 11 insertions(+), 1 deletion(-)

diff --git a/.gitlab-ci.yml b/.gitlab-ci.yml
index 230eb1fe1b..78321d4916 100644
--- a/.gitlab-ci.yml
+++ b/.gitlab-ci.yml
@@ -3,7 +3,17 @@ stages:
 
 .base_template : &BASE
    script:
-   - cat README.md
+   - nvidia-smi
+   - python -m pip install --upgrade pip
+   - pip install -r requirements.txt
+   - pip install flake8
+   - pip install pep8-naming
+   # stop the build if there are Python syntax errors or undefined names
+   - flake8 . --count --select=E9,F63,F7,F82 --show-source --statistics --config ./.flake8
+   # exit-zero treats all errors as warnings.
+   # - flake8 . --count --statistics --config ./.flake8
+   - ./runtests.sh --quick
+   - echo "Done with runtests.sh"
 
 build-ci-test:
     stage: build