github/tinygrad
1
1
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-01-09 15:08:02 -05:00
Code Issues Packages Projects Releases Wiki Activity

test kopt correctness (#1756)

Browse Source 
* test kopt correctness

* bump BUDGET to 20

* kopt hooks as setUp/tearDown
This commit is contained in:
nimlgen
2023-09-04 20:55:00 +03:00
committed by GitHub
parent c6d5d45a2b
commit f863c12610
3 changed files with 51 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -160,7 +160,7 @@ jobs:
PYTHONPATH="." python test/external/dist/test_collectives.py
- if: ${{ matrix.task == 'kopt' }}
name: Test KOPT
run: PYTHONPATH="." KOPT=1 BUDGET=10 GPU=1 DEBUG=1 python -m pytest -rA -n=auto test/models/test_real_world.py
run: PYTHONPATH="." KOPT=1 BUDGET=20 GPU=1 DEBUG=1 python -m pytest -rA -n=auto test/models/test_real_world.py
testmetalwebgpu:
name: Metal and WebGPU Tests