commaai
/
tinygrad
mirror of https://github.com/commaai/tinygrad.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Parallelize CI tests (#535)

This commit is contained in:
James Roberts 2023-02-06 23:27:44 +02:00 committed by GitHub
parent 4977d6f225
commit 0d405fd5bc
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 8 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
.github/workflows/test.yml vendored
View File

@ -56,9 +56,9 @@ jobs:
- name: Install Dependencies - name: Install Dependencies
run: pip install -e '.[testing]' --extra-index-url https://download.pytorch.org/whl/cpu run: pip install -e '.[testing]' --extra-index-url https://download.pytorch.org/whl/cpu
- name: Run Pytest - name: Run Pytest
run: LAZY=0 python -m pytest -s -v run: LAZY=0 python -m pytest -s -v -n=auto
- name: Run Pytest (lazy) - name: Run Pytest (lazy)
run: LAZY=1 python -m pytest -s -v run: LAZY=1 python -m pytest -s -v -n=auto
testllvm: testllvm:
name: LLVM Tests name: LLVM Tests
@ -74,7 +74,7 @@ jobs:
- name: Install Dependencies - name: Install Dependencies
run: pip install -e '.[llvm,testing]' --extra-index-url https://download.pytorch.org/whl/cpu run: pip install -e '.[llvm,testing]' --extra-index-url https://download.pytorch.org/whl/cpu
- name: Run Pytest OPT=2 - name: Run Pytest OPT=2
run: OPT=2 LLVM=1 python -m pytest -s -v run: OPT=2 LLVM=1 python -m pytest -s -v -n=auto
testtorch: testtorch:
name: Torch Tests name: Torch Tests
@ -90,9 +90,9 @@ jobs:
- name: Install Dependencies - name: Install Dependencies
run: pip install -e '.[testing]' --extra-index-url https://download.pytorch.org/whl/cpu run: pip install -e '.[testing]' --extra-index-url https://download.pytorch.org/whl/cpu
- name: Run Pytest - name: Run Pytest
run: LAZY=0 TORCH=1 python -m pytest -s -v run: LAZY=0 TORCH=1 python -m pytest -s -v -n=auto
- name: Run Pytest (lazy) - name: Run Pytest (lazy)
run: LAZY=1 TORCH=1 python -m pytest -s -v run: LAZY=1 TORCH=1 python -m pytest -s -v -n=auto
testgpu: testgpu:
name: GPU Tests name: GPU Tests
@ -118,7 +118,7 @@ jobs:
- name: Run Optimizer Test - name: Run Optimizer Test
run: OPT=2 GPU=1 python test/external_test_opt.py run: OPT=2 GPU=1 python test/external_test_opt.py
- name: Run Pytest (default) - name: Run Pytest (default)
run: OPT=1 GPU=1 python -m pytest -s -v run: OPT=1 GPU=1 python -m pytest -s -v -n=auto
testopencl: testopencl:
name: openpilot (OpenCL) Test name: openpilot (OpenCL) Test
@ -160,4 +160,4 @@ jobs:
- name: Build Docker - name: Build Docker
run: docker build -t tinygrad -f test/Dockerfile . run: docker build -t tinygrad -f test/Dockerfile .
- name: Test Docker - name: Test Docker
run: docker run --rm tinygrad /usr/bin/env python3 -c "from tinygrad.tensor import Tensor; print(Tensor.eye(3).numpy())" run: docker run --rm tinygrad /usr/bin/env python3 -c "from tinygrad.tensor import Tensor; print(Tensor.eye(3).numpy())"

1
setup.py
View File

@ -36,6 +36,7 @@ setup(name='tinygrad',
"torch~=1.13.0", "torch~=1.13.0",
"protobuf~=3.19.0", "protobuf~=3.19.0",
"pytest", "pytest",
"pytest-xdist",
"onnx~=1.12.0", "onnx~=1.12.0",
"onnx2torch", "onnx2torch",
], ],