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_shapetracker.py

add cache collector (#1595)

2023-08-28 19:59:55 -07:00

test_conv.py

fix wino mask upcast calculation (#2057)

2023-10-18 16:54:48 -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_failures.py

start work on auto opt (#2034)

2023-10-11 12:54:53 -07:00

test_linearizer.py

optimizer: simplify GROUP and LOCAL to have one of each (#2162)

2023-10-27 11:37:44 -10: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