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

* Enable JIT test

* really test metal

* Skip some device

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

..

external

Fix naming conflict with huggingface datasets (#1161)

2023-07-07 10:43:44 -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

Simplify symbolic.SumNode.__floordiv__ logic (#1220)

2023-07-12 12:54:12 -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_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

Fix: negative axis in tensor.cumsum (#1261)

2023-07-17 16:16: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