tinygrad/.pre-commit-config.yaml

repos:
  - repo: local
    hooks:
      - id: whitespace
        name: strip whitespace
        entry: find tinygrad -type f -name "*.py" -exec sed -i '' 's/ *$//' '{}' ';'
        language: system
        always_run: true
        pass_filenames: false
      - id: ruff
        name: ruff
        entry: ruff check .
        language: system
        always_run: true
        pass_filenames: false
      - id: mypy
        name: mypy
        entry: mypy tinygrad/ --strict-equality
        language: system
        always_run: true
        pass_filenames: false
      - id: docs2
        name: docs2
        entry: python3 docs/abstractions2.py
        language: system
        always_run: true
        pass_filenames: false
      - id: devicetests
        name: select GPU tests
        entry: env GPU=1 PYTHONPATH="." pytest test/test_uops.py test/test_search.py
        language: system
        always_run: true
        pass_filenames: false
      - id: tests
        name: subset of tests
        entry: env PYTHONPATH="." python3 -m pytest -n=4 test/unit/ test/test_ops.py test/test_dtype.py test/test_schedule.py test/test_assign.py test/test_symbolic_shapetracker.py
        language: system
        always_run: true
        pass_filenames: false
      - id: example
        name: multi device tests
        entry: python3 test/external/external_test_example.py
        language: system
        always_run: true
        pass_filenames: false
      - id: pylint
        name: pylint
        entry: env PYTHONPATH="." python3 -m pylint tinygrad/
        language: system
        always_run: true
        pass_filenames: false
Add commit hooks (#478) * Add pre-commit hook * We need ret * Fix some type definitions 2023-01-27 14:24:31 +08:00			`repos:`
fix shape types 2023-01-27 14:29:20 +08:00			`- repo: local`
			`hooks:`
strip whitespace 2023-11-26 06:09:06 +08:00			`- id: whitespace`
			`name: strip whitespace`
move disassemblers and openpilot (#4592) * move disassemblers and openpilot * delete junk * put that in pre-commit * fixup readme 2024-05-15 10:30:02 +08:00			`entry: find tinygrad -type f -name ".py" -exec sed -i '' 's/ $//' '{}' ';'`
strip whitespace 2023-11-26 06:09:06 +08:00			`language: system`
			`always_run: true`
			`pass_filenames: false`
move print tree into graph (#2003) * move print tree into graph * add winograd profiling test * change pre-commit to run ruff first 2023-10-07 19:39:21 +08:00			`- id: ruff`
			`name: ruff`
ruff cleanup (#4594) * check editor config * no editorconfig, it doesn't work * ruff cleanups 2024-05-15 12:16:14 +08:00			`entry: ruff check .`
move print tree into graph (#2003) * move print tree into graph * add winograd profiling test * change pre-commit to run ruff first 2023-10-07 19:39:21 +08:00			`language: system`
			`always_run: true`
			`pass_filenames: false`
fast path for copy (#2548) * fast copy * ruff first * flat_mv on malloc * order + webgpu test 2023-12-02 03:34:47 +08:00			`- id: mypy`
			`name: mypy`
add mypy --strict-equality to pre-commit (#3458) matched ci mypy behavior 2024-02-21 16:41:05 +08:00			`entry: mypy tinygrad/ --strict-equality`
fast path for copy (#2548) * fast copy * ruff first * flat_mv on malloc * order + webgpu test 2023-12-02 03:34:47 +08:00			`language: system`
			`always_run: true`
			`pass_filenames: false`
remove numpy from device (#3123) * remove numpy from device * fix tests * np item * cleanups * simplify with as_buffer * no toCPU * tinygradic * cast to scalar 2024-01-15 11:36:05 +08:00			`- id: docs2`
			`name: docs2`
docs cleanup and move (#4593) * cleanup and move * docs-legacy is gone * don't update setup.py 2024-05-15 11:44:59 +08:00			`entry: python3 docs/abstractions2.py`
refactor to remove extra kernel params (#2563) * refactor to have compiled kernel * bugfixes * docs/beautiful.py * revert that * fix tests 2023-12-02 16:32:25 +08:00			`language: system`
			`always_run: true`
			`pass_filenames: false`
			`- id: devicetests`
			`name: select GPU tests`
remove custom op support, we waste time maintaining this (#6991) * remove custom op support, we waste time maintaining this * customop is over 2024-10-11 14:31:09 +08:00			`entry: env GPU=1 PYTHONPATH="." pytest test/test_uops.py test/test_search.py`
Devicebufferless (#708) * runs one metal kernel * conv2d works * ops tests are passing * const folding * all ops work * pre commit always passes * torch works * working still * fix graph test * tests passing * image almost works * image conv works * most images * fix custom * fix assignment * fix compile enet * clean up comments * fix realize return value * include shapetracker in LB repr * copy should make a copy * reenable method cache * fix lna * dtypes in graph * forward only for IMAGE=2 * simple realize * getting close * fixup new api, it's good except the kernel count * back to 197 kernels * tests should pass * go to a real float * no type_on_cpu * fix the docs * put shapetracker back in it's proper place 2023-03-19 05:40:23 +08:00			`language: system`
			`always_run: true`
			`pass_filenames: false`
tests in pre commit 2023-03-13 13:42:26 +08:00			`- id: tests`
remove use of TORCH in pre-commit (#3424) it's silently using DEFAULT after removing TORCH 2024-02-16 08:38:37 +08:00			`name: subset of tests`
remove custom op support, we waste time maintaining this (#6991) * remove custom op support, we waste time maintaining this * customop is over 2024-10-11 14:31:09 +08:00			`entry: env PYTHONPATH="." python3 -m pytest -n=4 test/unit/ test/test_ops.py test/test_dtype.py test/test_schedule.py test/test_assign.py test/test_symbolic_shapetracker.py`
proper wmma (#2245) * proper wmma * hip cast * bugfixes * bugfix * that bug is fixed --------- Co-authored-by: George Hotz <george@tinygrad.org> 2023-11-10 07:15:18 +08:00			`language: system`
			`always_run: true`
			`pass_filenames: false`
			`- id: example`
			`name: multi device tests`
more test cleanups (#2631) * more test cleanups * move test example back 2023-12-06 08:17:57 +08:00			`entry: python3 test/external/external_test_example.py`
tests in pre commit 2023-03-13 13:42:26 +08:00			`language: system`
			`always_run: true`
			`pass_filenames: false`
remove arm64, caching for cuda (#2201) * remove arm64, caching for cuda * caching in llvm * switch cache_compiled to new cache * fix clang * caching for metal * fix pylint * cleanups * perf_counter and binary 2023-11-02 09:44:00 +08:00			`- id: pylint`
			`name: pylint`
move things, clean up extra (#2292) * move things * idk why pylint needs that now * delete unused 2023-11-14 12:18:40 +08:00			`entry: env PYTHONPATH="." python3 -m pylint tinygrad/`
remove arm64, caching for cuda (#2201) * remove arm64, caching for cuda * caching in llvm * switch cache_compiled to new cache * fix clang * caching for metal * fix pylint * cleanups * perf_counter and binary 2023-11-02 09:44:00 +08:00			`language: system`
			`always_run: true`
rename uop [run_process_replay] (#5031) * rename * fix unittests * rename vin * fix test * fix type [run_process_replay] * rm pre commit hook change 2024-06-19 02:34:05 +08:00			`pass_filenames: false`