add torch tests
@@ -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
