schedule2, keep the tests working with small changes (#1932)

* lazy cleanups

* ast functions take in LazyOps

* op instead of self.op

* _base for mops

* fix contiguous

* start schedule

* test_schedule

* fix openpilot

* more tests

* bugfix and test skip

* work

* make sure things get freed

* fix zerosized tensors

* fix failing test

* fix ceil and friends

* fix openpilot

* disable training

* disable test collectives
This commit is contained in:
George Hotz
2023-09-28 09:14:43 -07:00
committed by GitHub
parent c6d5e471d0
commit adab724caa
13 changed files with 491 additions and 96 deletions

View File

@@ -21,7 +21,7 @@ repos:
pass_filenames: false
- id: tests
name: subset of (CPU) tests
entry: env CPU=1 pytest test/unit/ test/test_ops.py test/test_dtype.py
entry: env CPU=1 pytest test/unit/ test/test_ops.py test/test_dtype.py test/test_schedule.py
language: system
always_run: true
pass_filenames: false