Rayan HatoutandGitHub 9975f24452 Fold expand preceding reduce if the reduction is on the same axis as the expansion (#1134) ...

* fold expands that precede a reduce if the reduction is on the same axis as the expansion

* add deterministic test for SIMPLIFY_SUM_RESHAPE_EXPAND_SUM optimization

* add a test case to make sure we don't fold reduce-expand-reduce on different axes

2023-07-06 13:41:05 -07:00

..

external

Fold expand preceding reduce if the reduction is on the same axis as the expansion (#1134)

2023-07-06 13:41:05 -07:00

extra

Make cross_process use cloudpickle (#1118)

2023-07-04 00:47:34 -07:00

models

Refactor nn.optim (#1091)

2023-07-02 15:07:30 -07:00

unit

from tensor cores + lb touchup (#1127)

2023-07-04 15:45:20 -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

Remove float64 (#1101)

2023-07-04 08:40:51 -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_net_speed.py

test speed llama (#1046)

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

test_nn.py

Windows & Ubuntu CLANG CI support (#1011)

2023-06-19 09:33:24 -07:00

test_ops.py

Remove POW llop and add SQRT llop (#1104)

2023-07-05 18:07:58 -07:00

test_optim.py

Remove POW llop and add SQRT llop (#1104)

2023-07-05 18:07:58 -07:00

test_randomness.py

Refactor LoadOps (#910)

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

test_specific_conv.py

disable that test on LLVM. i have to stop pushing to master

2023-05-26 03:11:03 +00:00

test_speed_v_torch.py

torch speed test, add add

2023-06-26 18:55:27 -07:00

test_tensor.py

Fix: Jacobian tests [WIP] (#1126)

2023-07-05 15:36:22 -07:00