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