George Hotz 81b73f97a3 Optiimzation (#355) ...

* constant folding into kernels

* that opt worth it?

* fix mypy

* ast one kernel

* save 2 lines in conv kernel

* debug print kernel count

* cl debugging

* early realize inputs

* refactor Device

2022-07-04 08:58:57 -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

graph_batchnorm.py

make contiguous ops yellow

2022-07-02 17:54:04 -07:00

test_conv.py

a little more readable

2022-06-27 08:54:04 -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

test lazy also, make TestMNIST faster

2022-07-03 15:19:19 -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

Optiimzation (#355)

2022-07-04 08:58:57 -07:00

test_ops.py

add test for slice_one

2022-07-03 12:14:20 -07:00

test_optim.py

Test split (#231)

2021-01-01 09:19:03 -05:00

test_shapetracker.py

simple shapetracker tests

2022-07-02 16:02:15 -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