From 6bee5bdb7d6a6acd09a34fc6fe2f66e845482ca7 Mon Sep 17 00:00:00 2001 From: George Hotz Date: Sat, 30 Oct 2021 18:58:45 -0700 Subject: [PATCH] add torch tests --- .github/workflows/test.yml | 20 ++++++++++++++++++-- 1 file changed, 18 insertions(+), 2 deletions(-) diff --git a/.github/workflows/test.yml b/.github/workflows/test.yml index b22960699a..637a0c2557 100644 --- a/.github/workflows/test.yml +++ b/.github/workflows/test.yml @@ -44,8 +44,6 @@ jobs: uses: actions/checkout@v2 - name: Update packages run: sudo apt-get update - - name: Install OpenCL - run: sudo apt-get install pocl-opencl-icd - name: Set up Python 3.8 uses: actions/setup-python@v2 with: @@ -55,6 +53,24 @@ jobs: - name: Run Pytest run: python -m pytest -s -v + testtorch: + name: Torch Tests + runs-on: ubuntu-latest + + steps: + - name: Checkout Code + uses: actions/checkout@v2 + - name: Update packages + run: sudo apt-get update + - name: Set up Python 3.8 + uses: actions/setup-python@v2 + with: + python-version: 3.8 + - name: Install Dependencies + run: pip install -e '.[testing]' + - name: Run Pytest + run: TORCH=1 python -m pytest -s -v + testgpu: name: GPU Tests runs-on: ubuntu-latest