Alexander Edwards 59af9b81c5 Match Torch speed for sum reduction on M1 (#1187) ...

* Add additional kernel when reducing multiple dimensions at once.

* Faster for smaller inputs

* Whitespace and naming

* Cleaner, guard for Metal only, and max 1 split rather than N

* Draft of different approach

* One additional kernel call for this test (as expected)

2023-07-19 09:18:58 -07:00

..

external

Match Torch speed for sum reduction on M1 (#1187)

2023-07-19 09:18:58 -07:00

extra

Fix: auto create parent dir when downloading file (#1173)

2023-07-07 13:40:29 -07:00

models

Webgpu support (#1077)

2023-07-12 12:52:06 -07:00

unit

Add Fuzz Test symbolic / shapetracker to CI. (#1278)

2023-07-19 09:05: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_assign.py

Refactor LoadOps (#910)

2023-06-03 09:40:43 -07:00

test_conv_shapetracker.py

speed like you wouldn't believe (#626)

2023-03-02 07:49:19 -08:00

test_conv.py

Add conv binops_no_rerun test assertions (#665)

2023-03-10 19:09:48 -08:00

test_custom_function.py

Optimizations in lazy.py (#987)

2023-06-26 13:55:42 -07:00

test_dtype.py

Webgpu support (#1077)

2023-07-12 12:52:06 -07:00

test_gc.py

fix up eye and fix gc test

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

test_jit.py

Enable JIT tests for supported devices, skip METAL and WEBGPU (#1265)

2023-07-18 11:40:37 -07:00

test_lazybuffer.py

Fix LazyBuffer SHUFFLE_PAD_OPS to prevent invalid pad movement (#1223)

2023-07-11 15:30:35 -07:00

test_linearizer.py

dedup kernel args v2 (#1272)

2023-07-18 20:03:42 -07:00

test_net_speed.py

test speed llama (#1046)

2023-06-25 15:22:56 -07:00

test_nn.py

Added nn.ConvTranspose1d (#1243)

2023-07-15 00:42:42 -07:00

test_ops.py

Slice migrated in Eye op (#1281)

2023-07-19 09:08:38 -07:00

test_optim.py

Over 90% on CIFAR with examples/hlb_cifar10.py (#1073)

2023-07-06 20:46:22 -07:00

test_randomness.py

Refactor LoadOps (#910)

2023-06-03 09:40:43 -07:00

test_specific_conv.py

Webgpu support (#1077)

2023-07-12 12:52:06 -07:00

test_speed_v_torch.py

Re-open create cumsum speed test (#1255)

2023-07-17 18:59:36 -07:00

test_tensor.py

Fix constant folding for Tensor([3]) (#1227)

2023-07-11 14:01:32 -07:00

test_webgpu.js

Webgpu support (#1077)

2023-07-12 12:52:06 -07:00