JIT support in Interpreted (#2314)

* factor that out

* jit is supported everywhere

* fix some tests

* there's no jit supported device, the jit is everywhere

* fix test uops
This commit is contained in:
George Hotz
2023-11-15 11:13:38 -08:00
committed by GitHub
parent 9a20bc08d6
commit 70a65c201e
16 changed files with 181 additions and 158 deletions

View File

@@ -1,6 +1,12 @@
repos:
- repo: local
hooks:
- id: mypy
name: mypy
entry: mypy tinygrad/ extra/helpers.py
language: system
always_run: true
pass_filenames: false
- id: ruff
name: ruff
entry: ruff .
@@ -19,12 +25,6 @@ repos:
language: system
always_run: true
pass_filenames: false
- id: mypy
name: mypy
entry: mypy tinygrad/ extra/helpers.py
language: system
always_run: true
pass_filenames: false
- id: tests
name: subset of TORCH tests
entry: env PYTHONPATH="." TORCH=1 python3 -m pytest -n=4 test/unit/ test/test_ops.py test/test_dtype.py test/test_schedule.py test/test_custom_function.py test/test_assign.py test/test_symbolic_shapetracker.py