tinygrad/test
George Hotz 643cbdfd50
make embedding and GPT-2 fast (#1631)
* make embedding fast

* jit more, variable shape support

* print mem bw
2023-08-22 15:14:38 -07:00
..
external make embedding and GPT-2 fast (#1631) 2023-08-22 15:14:38 -07:00
extra move state to nn/state (#1619) 2023-08-22 07:36:24 -07:00
models move state to nn/state (#1619) 2023-08-22 07:36:24 -07:00
unit move state to nn/state (#1619) 2023-08-22 07:36:24 -07:00
__init__.py All devices are equal! (#196) 2020-12-15 23:44:08 -08:00
Dockerfile Docker fix (#1039) 2023-06-25 10:38:58 -07:00
test_allocators.py init allocator for compiled backends (#1467) 2023-08-17 10:33:32 -07:00
test_assign.py Limit dims based on max size (#1390) 2023-07-31 19:18:19 -07:00
test_conv.py Limit dims based on max size (#1390) 2023-07-31 19:18:19 -07:00
test_conv_shapetracker.py CI < 5 minutes (#1252) 2023-07-23 13:00:56 -07:00
test_custom_function.py flake8 (#1323) 2023-07-24 11:19:58 -04:00
test_dtype.py Shorter (#1582) 2023-08-20 08:12:16 -07:00
test_gc.py fix up eye and fix gc test 2023-02-27 06:53:18 -08:00
test_helpers.py symbolic codegen and exec (#1552) 2023-08-16 14:43:41 -07:00
test_jit.py support symbols in shrink (#1611) 2023-08-22 09:08:21 -07:00
test_lazybuffer.py ShapeTracker.var_vals (#1540) 2023-08-14 18:53:37 -07:00
test_linearizer.py CSE at uop level (#1483) 2023-08-19 23:40:40 -07:00
test_net_speed.py CI < 5 minutes (#1252) 2023-07-23 13:00:56 -07:00
test_nn.py sparse cat cross entropy (#1597) 2023-08-21 14:14:54 -07:00
test_ops.py add noalias to llvm (#1622) 2023-08-22 09:26:01 -07:00
test_optim.py Add WEBGPU tests to CI (#1463) 2023-08-06 10:32:01 -07:00
test_randomness.py .cpu().numpy() -> .numpy() (#1594) 2023-08-21 09:53:29 -07:00
test_specific_conv.py CI < 5 minutes (#1252) 2023-07-23 13:00:56 -07:00
test_speed_v_torch.py add cpu if torch tensor (#1609) 2023-08-21 16:57:59 -07:00
test_symbolic_jit.py support symbols in shrink (#1611) 2023-08-22 09:08:21 -07:00
test_symbolic_ops.py support symbols in shrink (#1611) 2023-08-22 09:08:21 -07:00
test_symbolic_shapetracker.py support symbols in shrink (#1611) 2023-08-22 09:08:21 -07:00
test_tensor.py .cpu().numpy() -> .numpy() (#1594) 2023-08-21 09:53:29 -07:00
test_uops.py make TestNonFloatUOps.test_mul_bool pass on all platforms (#1557) 2023-08-16 11:34:09 -07:00
test_webgpu.js Revert "update editorconfig, enforce via CI (#1343)" (#1380) 2023-07-31 10:35:50 -07:00