graph rewrite tests (#6519)

* more graph rewrite tests

* more complex test cases

* more tests

* more tests

* cleanups

* 9600 lines

* cleanups
This commit is contained in:
George Hotz
2024-09-15 17:29:16 +08:00
committed by GitHub
parent 89b950c6b3
commit cd90092f14
3 changed files with 167 additions and 3 deletions

View File

@@ -144,8 +144,8 @@ jobs:
mypy -c "from tinygrad.tensor import Tensor; print(Tensor([1,2,3,4,5]))"
- name: Test DEBUG
run: DEBUG=100 python3 -c "from tinygrad import Tensor; N = 1024; a, b = Tensor.rand(N, N), Tensor.rand(N, N); c = (a.reshape(N, 1, N) * b.T.reshape(1, N, N)).sum(axis=2); print((c.numpy() - (a.numpy() @ b.numpy())).mean())"
- name: Repo line count <9500 lines
run: MAX_LINE_COUNT=9500 python sz.py
- name: Repo line count <9600 lines
run: MAX_LINE_COUNT=9600 python sz.py
testopencl:
strategy: