add torch tests

This commit is contained in:
George Hotz
2021-10-30 18:58:45 -07:00
parent 75260c6e49
commit 6bee5bdb7d

View File

@@ -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