mirror of
https://github.com/tinygrad/tinygrad.git
synced 2026-01-09 15:08:02 -05:00
add torch tests
This commit is contained in:
20
.github/workflows/test.yml
vendored
20
.github/workflows/test.yml
vendored
@@ -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
|
||||
|
||||
Reference in New Issue
Block a user