David Hou 3151d91f6e 3x3 winograd convs (#1675) ...

* winograd

* simplify local groups code

* comment

* respects self.opts.has_local

* always simplify ones

* make mypy happy

* move reshape, WINO flag

* wino flag, simple forward backward test for wino

* extra wino test

* merge oops

* comments

* axis_needs_valid -> axis_is_masked

* don't delete needs_valid (it's unused though)

* make linter happy

* make linter happy

* smaller test

* change number

* make wino tests very small

2023-09-03 07:29:43 -07:00

..

external

codellama (#1702)

2023-09-02 08:45:12 -07:00

extra

[ready] Replacing os with pathlib (#1708)

2023-08-30 10:41:08 -07:00

models

faster gpt2 jit path and gpt2 in test_real_world (#1738)

2023-09-02 08:39:12 -07:00

unit

visitor pattern for sym_infer and unit tests (#1733)

2023-09-01 09:47:45 -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

allow zerosized tensors (#1659)

2023-08-30 10:39:24 -07:00

test_assign.py

Limit dims based on max size (#1390)

2023-07-31 19:18:19 -07:00

test_cache_collector.py

add cache collector (#1595)

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

test_conv_shapetracker.py

add cache collector (#1595)

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

test_conv.py

3x3 winograd convs (#1675)

2023-09-03 07:29:43 -07:00

test_custom_function.py

Optional: Reduce line count and simplify the LazyBuffer interface (#1642)

2023-08-22 21:01:10 -07:00

test_dtype.py

move device to ops (#1646)

2023-08-23 08:30:17 -07:00

test_gc.py

fix up eye and fix gc test

2023-02-27 06:53:18 -08:00

test_helpers.py

add dtypes test (#1682)

2023-08-28 08:12:15 -07:00

test_jit.py

support symbols in shrink (#1611)

2023-08-22 09:08:21 -07:00

test_lazybuffer.py

add cache collector (#1595)

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

test_linearizer.py

add cache collector (#1595)

2023-08-28 19:59:55 -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

Make __getitem__ TINYer (#1661)

2023-09-02 23:01:01 -04:00

test_optim.py

Add WEBGPU tests to CI (#1463)

2023-08-06 10:32:01 -07:00

test_randomness.py

Tensor.uniform with dtype=int bug fix (#1593)

2023-08-26 01:59:53 -04:00

test_specific_conv.py

move device to ops (#1646)

2023-08-23 08:30:17 -07:00

test_speed_v_torch.py

move device to ops (#1646)

2023-08-23 08:30:17 -07:00

test_symbolic_jit.py

llama JIT python runtime speedup (#1633)

2023-08-30 07:51:05 -07:00

test_symbolic_ops.py

fix symbolic_ops tests with Tensor.training=True (#1686)

2023-08-26 23:19:56 -04:00

test_symbolic_shapetracker.py

support symbols in shrink (#1611)

2023-08-22 09:08:21 -07:00

test_tensor.py

allow zerosized tensors (#1659)

2023-08-30 10:39:24 -07:00

test_uops.py

render const (#1736)

2023-09-01 19:01:43 -07:00

test_webgpu.js

Revert "update editorconfig, enforce via CI (#1343)" (#1380)

2023-07-31 10:35:50 -07:00