Parallelize CI tests
MODIFY
.github/workflows/test.yml
.github/workflows/test.yml
@@ -56,9 +56,9 @@ jobs:
- name: Install Dependencies
run: pip install -e '.[testing]' --extra-index-url https://download.pytorch.org/whl/cpu
- 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)
- run: LAZY=1 python -m pytest -s -v
+ run: LAZY=1 python -m pytest -s -v -n=auto
testllvm:
name: LLVM Tests
@@ -74,7 +74,7 @@ jobs:
- name: Install Dependencies
run: pip install -e '.[llvm,testing]' --extra-index-url https://download.pytorch.org/whl/cpu
- 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:
name: Torch Tests
@@ -90,9 +90,9 @@ jobs:
- name: Install Dependencies
run: pip install -e '.[testing]' --extra-index-url https://download.pytorch.org/whl/cpu
- 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)
- run: LAZY=1 TORCH=1 python -m pytest -s -v
+ run: LAZY=1 TORCH=1 python -m pytest -s -v -n=auto
testgpu:
name: GPU Tests
@@ -118,7 +118,7 @@ jobs:
- name: Run Optimizer Test
run: OPT=2 GPU=1 python test/external_test_opt.py
- 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:
name: openpilot (OpenCL) Test

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

Parallelize CI tests (#535)