George Hotz a5820390db All LazyOps in the Linearizer (#1905) ...

* loadop buffer on cpu

* works for GPU

* sort of working

* has bugs

* gpu tests pass

* fix some tests

* fix tensor cores

* fix test linearizer

* fix symbolic

* fix has_variable_shape

* non symbolic size

* disable weird test

* simple cache fix

* fix custom function

* fix kopt

* cleanups

* a bit broken on the assign

* contig check

* only buffer

* need that order

* idx

2023-09-24 11:50:00 +08:00

..

external

Add needed contiguous to DiskBuffer. SHM support on OSX (#1891)

2023-09-22 09:16:42 +08:00

extra

fixes (#1893)

2023-09-22 07:20:27 +08:00

models

All LazyOps in the Linearizer (#1905)

2023-09-24 11:50:00 +08:00

unit

All LazyOps in the Linearizer (#1905)

2023-09-24 11:50:00 +08: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_allocators.py

no import when Python is shutting down (#1875)

2023-09-20 12:47:02 -04:00

test_assign.py

patch to remove hack from stable_diffusion.py (#1814)

2023-09-08 09:26:50 -07:00

test_cache_collector.py

no functions with same names in test/ (#1811)

2023-09-07 11:27:31 -07:00

test_conv_shapetracker.py

add cache collector (#1595)

2023-08-28 19:59:55 -07:00

test_conv.py

3x3 winograd convs (#1675)

2023-09-03 07:29:43 -07:00

test_custom_function.py

All LazyOps in the Linearizer (#1905)

2023-09-24 11:50:00 +08:00

test_dtype.py

move device to ops (#1646)

2023-08-23 08:30:17 -07:00

test_gc.py

fix up eye and fix gc test

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

test_jit.py

support symbols in shrink (#1611)

2023-08-22 09:08:21 -07:00

test_lazybuffer.py

Apply ShapeTracker in interpreted backends (#1846)

2023-09-23 10:05:13 +08:00

test_linearizer.py

All LazyOps in the Linearizer (#1905)

2023-09-24 11:50:00 +08:00

test_net_speed.py

CI < 5 minutes (#1252)

2023-07-23 13:00:56 -07:00

test_nn.py

3x3 winograd convs (#1675)

2023-09-03 07:29:43 -07:00

test_ops.py

Make __getitem__ TINYer (#1661)

2023-09-02 23:01:01 -04:00

test_optim.py

use class Foo: instead of class Foo(): (#1797)

2023-09-06 12:20:25 -07:00

test_randomness.py

Tensor.uniform with dtype=int bug fix (#1593)

2023-08-26 01:59:53 -04:00

test_specific_conv.py

move device to ops (#1646)

2023-08-23 08:30:17 -07:00

test_speed_v_torch.py

linearizer: support local and group_for_reduce dimensions together (#1821)

2023-09-08 12:39:27 -07:00

test_symbolic_jit.py

Move var_vals from ShapeTracker to LazyBuffer (#1819)

2023-09-08 09:25:10 -07:00

test_symbolic_ops.py

Enable symbolic ops tests for LLVM (#1898)

2023-09-23 07:30:26 +08:00

test_symbolic_shapetracker.py

symbolic shape type with TypeGuard (#1852)

2023-09-13 05:27:22 +08:00

test_tensor.py

add test_tensor_copy (#1840)

2023-09-10 16:01:58 -07:00

test_uops.py

Make Triton work again (#1547)

2023-09-23 14:17:12 +08:00

test_webgpu.js

Revert "update editorconfig, enforce via CI (#1343)" (#1380)

2023-07-31 10:35:50 -07:00