tinygrad/test
Ahmed Harmouche 2114dc13d1
Allow multi-input model export (#1995)
* Allow multi-input model export

* Add model export unit test

* Fix efficientnet compilation

* Only run model export test on JIT supported devices

* Skip export model test if not EXPORT_SUPPORTED_DEVICE
2023-10-07 04:13:34 -07:00
..
external remove stale tests 2023-10-06 02:14:56 -07:00
extra Allow multi-input model export (#1995) 2023-10-07 04:13:34 -07:00
models Fix llama with empty prompt (#1997) 2023-10-06 06:48:07 -07:00
unit good changes from openpilot_compile2 (#2000) 2023-10-06 13:33:24 -07:00
Dockerfile Docker fix (#1039) 2023-06-25 10:38:58 -07:00
__init__.py
test_allocators.py no import when Python is shutting down (#1875) 2023-09-20 12:47:02 -04:00
test_assign.py corealize + remove realize from lazybuffer (#1968) 2023-10-04 10:59:31 -07:00
test_cache_collector.py fix jitted dist (#1955) 2023-10-02 11:45:13 -04:00
test_conv.py 3x3 winograd convs (#1675) 2023-09-03 07:29:43 -07:00
test_conv_shapetracker.py add cache collector (#1595) 2023-08-28 19:59:55 -07:00
test_custom_function.py unrealized consts everywhere (#1963) 2023-10-04 01:48:10 -07:00
test_dtype.py Fix cast error in render_load in wgsl (#1956) 2023-10-04 02:29:14 -07:00
test_gc.py fix up eye and fix gc test 2023-02-27 06:53:18 -08:00
test_jit.py unrealized consts everywhere (#1963) 2023-10-04 01:48:10 -07:00
test_lazybuffer.py corealize + remove realize from lazybuffer (#1968) 2023-10-04 10:59:31 -07:00
test_linearizer.py no process() in Linearizer (#1966) 2023-10-04 07:18:42 -07:00
test_net_speed.py CI < 5 minutes (#1252) 2023-07-23 13:00:56 -07:00
test_nn.py 3x3 winograd convs (#1675) 2023-09-03 07:29:43 -07:00
test_ops.py schedule2, keep the tests working with small changes (#1932) 2023-09-28 09:14:43 -07:00
test_optim.py use class Foo: instead of class Foo(): (#1797) 2023-09-06 12:20:25 -07:00
test_randomness.py Tensor.uniform with dtype=int bug fix (#1593) 2023-08-26 01:59:53 -04:00
test_schedule.py good changes from openpilot_compile2 (#2000) 2023-10-06 13:33:24 -07:00
test_specific_conv.py move device to ops (#1646) 2023-08-23 08:30:17 -07:00
test_speed_v_torch.py wmma: add test and tensor core shape (#1925) 2023-09-28 18:04:28 -07:00
test_symbolic_jit.py Move var_vals from ShapeTracker to LazyBuffer (#1819) 2023-09-08 09:25:10 -07:00
test_symbolic_ops.py with Tensor.train() (#1935) 2023-09-28 18:02:31 -07:00
test_symbolic_shapetracker.py Make ShapeTracker Immutable (#1909) 2023-09-24 21:09:03 +08:00
test_tensor.py fix transfer to mapped buffers (#1923) 2023-09-29 00:50:24 -07:00
test_uops.py Make Triton work again (#1547) 2023-09-23 14:17:12 +08:00
test_webgpu.js Revert "update editorconfig, enforce via CI (#1343)" (#1380) 2023-07-31 10:35:50 -07:00
test_winograd.py winograd test prints op count 2023-09-29 05:41:29 -07:00