tinygrad/test/models
George Hotz 3b0b3a2e64
fast RANGEIFY (#12504)
* rtoposort is fast, can replace rangeify with this

* fast rangeify

* work

* fast rangeify works for mnist

* should work

* progress

* pad fix

* FAST

* tests passing

* don't delete those shape ops

* put in rangeify map

* ending ranges fix

* tests

* mstack/mselect no hacks

* move to indexing.py

* touch up tests + add comments

* disable failing test

* actually make the file readable

* failing

* error
2025-10-08 19:38:06 +08:00
..
efficientnet test cleanups 2023-03-03 06:36:06 -08:00
waifu2x waifu2x vgg7: testcase, auto-RGBA->RGB, function to grab pretrained models, training "fix" (#2117) 2023-10-19 22:07:15 -07:00
whisper whisper: support batch inference, add librispeech WER test (#2074) 2023-11-16 13:50:08 -08:00
__init__.py more test cleanups (#2631) 2023-12-05 16:17:57 -08:00
test_bert.py update test/models (#12142) 2025-09-12 16:43:28 -04:00
test_efficientnet.py update test_chicken (#12466) 2025-10-06 03:58:44 -04:00
test_end2end.py tests from grad uop path [pr] (#8313) 2024-12-18 09:25:05 -08:00
test_mnist.py CLANG -> CPU (#9189) 2025-02-20 18:03:09 -05:00
test_onnx.py move frontend dir to nn [pr] (#12470) 2025-10-07 10:42:22 +08:00
test_real_world.py fast RANGEIFY (#12504) 2025-10-08 19:38:06 +08:00
test_resnet.py Fix FC layer ResNet load_from_pretrained error (#8387) 2024-12-26 18:11:27 -05:00
test_rnnt.py update test/models (#12142) 2025-09-12 16:43:28 -04:00
test_train.py update test/models (#12142) 2025-09-12 16:43:28 -04:00
test_waifu2x.py Apply ruff linting rules to tests (#2473) 2023-11-27 21:24:06 -08:00
test_whisper.py unify cpu and llvm (#11982) 2025-09-09 13:54:44 +03:00