Stan 264d467f2b Added tensor.squeeze and support for testing exceptions (#1241) ...

* WIP: `tensor.squeeze` function

* Added `test_except` param to `helper_test_op` to avoid false positives

* Extracted new method `helper_test_exception` for testing exceptions

* Made `squeeze` not throw IndexError when ndim == 0 and dim <= 0 to match PyTorch

2023-07-15 00:33:24 -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

assertions for jit

2023-05-05 21:56:32 -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 test for nn.Conv1d (#1242)

2023-07-15 00:30:50 -07:00

test_ops.py

Added tensor.squeeze and support for testing exceptions (#1241)

2023-07-15 00:33:24 -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

Webgpu support (#1077)

2023-07-12 12:52:06 -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