mirror of
https://github.com/tinygrad/tinygrad.git
synced 2026-06-24 02:14:17 +00:00
ebd72ff437* Split tests Split tests into "Test CPU" and "Test GPU". Add test flag "TEST_DEVICES" which is a comma separated list of devices: CPU,GPU,ANE * Run tests based on provided TEST_DEVICES flag By default will run all "CPU,GPU,ANE" * fix bad quote * Revert changes and use GPU=1 This is done through setting the default Tensor Device to Device.CPU of GPU=1 is set. Run GPU tests: GPU=1 pytest -s -v
75 lines
1.9 KiB
YAML
75 lines
1.9 KiB
YAML
name: Unit Tests
|
|
|
|
on: [push, pull_request]
|
|
|
|
jobs:
|
|
lines:
|
|
name: Less than 1000 lines
|
|
runs-on: ubuntu-latest
|
|
|
|
steps:
|
|
- name: Checkout Code
|
|
uses: actions/checkout@v2
|
|
- name: Install SLOCCount
|
|
run: sudo apt-get install sloccount
|
|
- name: Check <1000 lines
|
|
run: sloccount tinygrad test examples; if [ $(sloccount tinygrad | sed -n 's/.*Total Physical Source Lines of Code (SLOC)[ ]*= \([^ ]*\).*/\1/p' | tr -d ',') -gt 1000 ]; then exit 1; fi
|
|
|
|
linter:
|
|
name: Indentation Linter
|
|
runs-on: ubuntu-latest
|
|
|
|
steps:
|
|
- name: Checkout Code
|
|
uses: actions/checkout@v2
|
|
- name: Set up Python 3.8
|
|
uses: actions/setup-python@v2
|
|
with:
|
|
python-version: 3.8
|
|
- name: Install dependencies
|
|
run: |
|
|
python -m pip install --upgrade pip
|
|
python -m pip install pylint
|
|
# pip install -r requirements.txt
|
|
- name: Lint with pylint
|
|
run: |
|
|
python -m pylint --disable=all -e W0311 --jobs=0 --indent-string=' ' **/*.py
|
|
|
|
testcpu:
|
|
name: CPU Tests
|
|
runs-on: ubuntu-latest
|
|
|
|
steps:
|
|
- name: Checkout Code
|
|
uses: actions/checkout@v2
|
|
- name: Install OpenCL
|
|
run: sudo apt-get install pocl-opencl-icd
|
|
- 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: python -m pytest -s -v
|
|
|
|
testgpu:
|
|
name: GPU Tests
|
|
runs-on: ubuntu-latest
|
|
|
|
steps:
|
|
- name: Checkout Code
|
|
uses: actions/checkout@v2
|
|
- name: Install OpenCL
|
|
run: sudo apt-get install pocl-opencl-icd
|
|
- name: Set up Python 3.8
|
|
uses: actions/setup-python@v2
|
|
with:
|
|
python-version: 3.8
|
|
- name: Install Dependencies
|
|
run: pip install -e '.[gpu,testing]'
|
|
- name: Run Pytest
|
|
run: GPU=1 python -m pytest -s -v
|
|
|
|
|
|
|