tinygrad/test
George Hotz 159a2d1a80
Simple Lazy (#340)
* simple lazy

* simple

* fix graph and make realize simpler

* SHUFFLE_MOVEMENT_OPS already works

* MERGE_MOVEMENT_OPS and REMOVE_MOVEMENT_NOPS

* it works, but it's slow

* constant inlining

* cache misses are the reason for loss

* fix non determinism

* cleanup, a few tests fail

* profile

* cache lazyop

* cleanups

* create namedtuple once

* bunch of caches

* it's not deleting

* nograd

* caching allocator

* reduce_op

* fromCPU if you want fromCPU

* complain

* nvidia fix

* realized on Tensor

* numpy is very slow

* no loads in second run

* caching in View

* 10ms speedups on batman

* remove old profiler

* bunch of refactors

* contiguous on view

* elementwise_op_compile for conv

* support ewop after processing op

* this still works

* conv folding works

* all we do is conv conv conv no matter what

* all args to the conv

* still works

* unify conv and ewop

* ops_gpu cleanup

* move around ops_gpu

* remove caching allocator

* remove unused

* find_conv shorten

* gpu refactors

* simpler gpu

* and that

* cmp is fast

* 18ms on mac

* it's a lot of lines, but it's faster

* minor

* tests pass

* LoadOps.CONTIGUOUS

* remove dups

* torch converter doesn't support slice

* move lazy out for merge

* LoadOps are only for lazy
2022-06-20 22:45:11 -07:00
..
efficientnet add ViT test and car 2022-06-05 17:12:43 -07:00
__init__.py All devices are equal! (#196) 2020-12-15 23:44:08 -08:00
test_conv.py fix test 2022-06-19 17:07:09 -07:00
test_efficientnet.py test_onnx works with enet also 2022-06-11 14:30:26 -07:00
test_gc.py Test split (#231) 2021-01-01 09:19:03 -05:00
test_mnist.py sgd threestep graph is so pretty 2022-06-06 09:45:37 -07:00
test_net_speed.py Test split (#231) 2021-01-01 09:19:03 -05:00
test_nn.py fix bn test, affine is True 2022-01-15 19:52:15 -08:00
test_onnx.py Simple Lazy (#340) 2022-06-20 22:45:11 -07:00
test_ops.py touchups 2022-06-19 16:57:14 -07:00
test_optim.py Test split (#231) 2021-01-01 09:19:03 -05:00
test_shapetracker.py Non contig (#339) 2022-06-19 22:40:48 -07:00
test_tensor.py skip broken tests, no float64 allowed 2022-06-11 17:12:04 -07:00
test_train.py the big memory gradient didn't even need to be computed 2022-06-16 11:41:29 -07:00