tinygrad/test/models
George Hotz a5820390db
All LazyOps in the Linearizer (#1905)
* loadop buffer on cpu

* works for GPU

* sort of working

* has bugs

* gpu tests pass

* fix some tests

* fix tensor cores

* fix test linearizer

* fix symbolic

* fix has_variable_shape

* non symbolic size

* disable weird test

* simple cache fix

* fix custom function

* fix kopt

* cleanups

* a bit broken on the assign

* contig check

* only buffer

* need that order

* idx
2023-09-24 11:50:00 +08:00
..
efficientnet test cleanups 2023-03-03 06:36:06 -08:00
test_bert.py move device to ops (#1646) 2023-08-23 08:30:17 -07:00
test_efficientnet.py Inference test: add tests for ResNet50 (#773) 2023-05-13 21:18:15 -07:00
test_end2end.py move state to nn/state (#1619) 2023-08-22 07:36:24 -07:00
test_mnist.py move state to nn/state (#1619) 2023-08-22 07:36:24 -07:00
test_onnx.py update openpilot supercombo.onnx to 0.9.4 (#1681) 2023-08-26 19:16:08 -04:00
test_real_world.py All LazyOps in the Linearizer (#1905) 2023-09-24 11:50:00 +08:00
test_rnnt.py .cpu().numpy() -> .numpy() (#1594) 2023-08-21 09:53:29 -07:00
test_train.py move state to nn/state (#1619) 2023-08-22 07:36:24 -07:00