tinygrad/test
Francis Lam 8cf0bb9351
optimizer: simplify GROUP and LOCAL to have one of each (#2162)
* optimizer: simplify GROUP and LOCAL to have one of each

Now that tensor cores only use LASTLOCAL, we can simplify to use
only that op everywhere.

The only use of GROUP is in matvec hand-coded opts and it doesn't
make a performance difference so switching to use only the top
behavior.

Also adds additional asserts to prevent tensor core dims from
being altered which causes bad kernels to be generated.

* search: remove duplicated actions
2023-10-27 11:37:44 -10:00
..
external perf: lazyop as dataclass (#1603) 2023-10-25 17:54:30 -04:00
extra Allow multi-input model export (#1995) 2023-10-07 04:13:34 -07:00
models perf: lazyop as dataclass (#1603) 2023-10-25 17:54:30 -04:00
unit Add dictionary keys to reduce db size (#2131) 2023-10-24 10:49:22 -04: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
helpers.py perf: lazyop as dataclass (#1603) 2023-10-25 17:54:30 -04:00
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 fix wino mask upcast calculation (#2057) 2023-10-18 16:54:48 -07:00
test_conv_shapetracker.py add cache collector (#1595) 2023-08-28 19:59:55 -07:00
test_custom_function.py Variable.bind newer (#2017) 2023-10-10 10:03:01 -07:00
test_dtype.py fix image dtype cmp (#2089) 2023-10-16 17:52:38 -07:00
test_exec.py Refactor hipgraph (#2141) 2023-10-24 15:45:56 -04: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_kernel_cache.py kernel cache (#2035) 2023-10-13 06:32:01 -07:00
test_lazybuffer.py corealize + remove realize from lazybuffer (#1968) 2023-10-04 10:59:31 -07:00
test_lazyop.py Q network for pruning BEAM / uops deduping / BEAM_ESTIMATE (#2142) 2023-10-27 10:53:06 -10:00
test_linearizer.py optimizer: simplify GROUP and LOCAL to have one of each (#2162) 2023-10-27 11:37:44 -10:00
test_linearizer_failures.py start work on auto opt (#2034) 2023-10-11 12:54:53 -07:00
test_net_speed.py CI < 5 minutes (#1252) 2023-07-23 13:00:56 -07:00
test_nn.py fix wino mask upcast calculation (#2057) 2023-10-18 16:54:48 -07:00
test_ops.py split test_conv2d 2023-10-18 14:00:50 -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 set default to standard uniform (#2158) 2023-10-27 16:15:30 -04:00
test_schedule.py add mops to graph, refactor IMAGE (#2100) 2023-10-17 21:27:51 -07:00
test_search.py add test coverage for search (#2104) 2023-10-19 17:06:47 -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 Variable.bind newer (#2017) 2023-10-10 10:03:01 -07:00
test_symbolic_ops.py Variable.bind newer (#2017) 2023-10-10 10:03:01 -07:00
test_symbolic_shapetracker.py Variable.bind newer (#2017) 2023-10-10 10:03:01 -07:00
test_tensor.py added from unaligned np test (#2134) 2023-10-23 11:38:57 -04:00
test_uops.py print test durations and add speed (#2107) 2023-10-18 13:46:42 -07:00
test_webgpu.js Revert "update editorconfig, enforce via CI (#1343)" (#1380) 2023-07-31 10:35:50 -07:00
test_winograd.py print test durations and add speed (#2107) 2023-10-18 13:46:42 -07:00