uops loop removal (#2262)

* remove the loop

* cleanups

* tests failing still

* global_loop_ctx wasn't needed

* replace_op is cleaner

* minor opt

* cast opt was wrong

* uop_num

* uop num was dumb

* tuplize_uops

* torch tests

* fix test_uops
This commit is contained in:
George Hotz
2023-11-10 15:24:47 -08:00
committed by GitHub
parent a753c8e071
commit 85d26ddc36
7 changed files with 97 additions and 63 deletions

View File

@@ -26,8 +26,8 @@ repos:
always_run: true
pass_filenames: false
- id: tests
name: subset of (CPU) tests
entry: env PYTHONPATH="." CPU=1 pytest 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
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
language: system
always_run: true
pass_filenames: false