1
0
Fork 0
Commit Graph

3 Commits (e6f19d4ce2a072d3dc3ba95c80c21c552f1753c8)

Author SHA1 Message Date
George Hotz d854337f0d nn/optim.py compiles now 2023-02-08 11:25:18 -06:00
George Hotz 1029deccb1 refactor ops_cpu and ops_torch to not share code 2023-02-08 11:11:42 -06:00
George Hotz 2844482a60
Mypy fun (#541)
* mypy fun

* things are just faster

* running fast

* mypy is fast

* compile.sh

* no gpu hack

* refactor ops_cpu and ops_torch to not subclass

* make weak buffer work

* tensor works

* fix test failing

* cpu/torch cleanups

* no or operator on dict in python 3.8

* that was junk

* fix warnings

* comment and touchup
2023-02-08 09:56:51 -06:00