George HotzandGitHub ca77d6cd72 bfloat16 in LLVM (enough for llama 2) (#1293) ...

* add bf16 support to LLVM

* bf16 read works

2023-07-19 20:18:32 -07:00

..

external

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

2023-07-19 13:39:16 -07:00

extra

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

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

models

add cifar training regression test (#1287)

2023-07-19 14:17:09 -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

bfloat16 in LLVM (enough for llama 2) (#1293)

2023-07-19 20:18:32 -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 test for size 1 tensor (#1285)

2023-07-19 11:06:40 -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

Added test_chunk and fixed (#1283)

2023-07-19 22:21:26 -04: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

torch test touchup

2023-07-19 09:37:23 -07:00

test_tensor.py

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

2023-07-19 10:51:06 -07:00

test_webgpu.js

Webgpu support (#1077)

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