George Hotz 8e9cdef61f clean up the buffers (#2447) ...

* clean up the buffers

* remove allocate_output

* functools.lru_cache is methodcache

* add TestShapeTrackerSize

* cache_clear

* no 0 sz buffer, add _ on functions that shouldn't be imported

* fix size

* if -> while

2023-11-26 11:02:29 -08:00

..

external

clean up the buffers (#2447)

2023-11-26 11:02:29 -08:00

extra

minor cleanups, remove dead files (#2398)

2023-11-23 09:01:50 -08:00

models

coder.py can write and run code (#2439)

2023-11-25 12:27:54 -08:00

unit

clean up the buffers (#2447)

2023-11-26 11:02:29 -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

helpers.py

perf: lazyop as dataclass (#1603)

2023-10-25 17:54:30 -04:00

test_allocators.py

Assert for memory allocation failures (#2337)

2023-11-16 20:14:16 -08:00

test_assign.py

move optimize_local_size (#2221)

2023-11-05 21:00:52 -08:00

test_conv_shapetracker.py

fix some tests related to JitItem (#2279)

2023-11-11 23:00:35 -05:00

test_conv.py

fix wino mask upcast calculation (#2057)

2023-10-18 16:54:48 -07:00

test_copy_speed.py

minor cleanups, remove dead files (#2398)

2023-11-23 09:01:50 -08:00

test_custom_function.py

JIT support in Interpreted (#2314)

2023-11-15 11:13:38 -08:00

test_dtype.py

From teeny (#2426)

2023-11-24 12:50:56 -08:00

test_gc.py

fix up eye and fix gc test

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

test_jit.py

From teeny (#2426)

2023-11-24 12:50:56 -08:00

test_kernel_cache.py

move all to compile api (#2203)

2023-11-01 23:01:32 -07:00

test_lazybuffer.py

examples of new GPT2 and JIT change (#2261)

2023-11-10 15:07:02 -05:00

test_lazyop.py

make self referential AST fast too (#2278)

2023-11-15 07:12:07 -08:00

test_linearizer_failures.py

test_linearizer_failures to run on all backends (#2443)

2023-11-26 01:17:29 -05:00

test_linearizer.py

remove force_wait, refactor to graph (#2405)

2023-11-23 12:46:07 -08:00

test_net_speed.py

Profiling-helper (#2321)

2023-11-16 14:15:56 -08:00

test_nn.py

A beautiful MNIST training example (#2272)

2023-11-17 19:42:43 -08:00

test_ops.py

From teeny (#2426)

2023-11-24 12:50:56 -08:00

test_optim.py

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

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

test_randomness.py

Pretty multinomial (#2365)

2023-11-19 15:10:10 -05:00

test_renderer.py

Move HIP render logic to its dedicated place (#2394)

2023-11-23 13:03:29 -08:00

test_sample.py

remove real_offset (#2234)

2023-11-07 17:30:53 -08:00

test_schedule.py

move things, clean up extra (#2292)

2023-11-13 20:18:40 -08:00

test_search.py

simple runtime args (#2211)

2023-11-03 12:31:29 -07:00

test_specific_conv.py

move device to ops (#1646)

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

test_speed_v_torch.py

move optimize_local_size (#2221)

2023-11-05 21:00:52 -08:00

test_symbolic_jit.py

From teeny (#2426)

2023-11-24 12:50:56 -08:00

test_symbolic_ops.py

fix the variable arg order (#2382)

2023-11-21 12:02:31 -05:00

test_symbolic_shapetracker.py

From teeny (#2426)

2023-11-24 12:50:56 -08:00

test_tensor.py

update some test_tensor.py cases with 0 in shape (#2368)

2023-11-19 20:35:05 -05:00

test_to_numpy.py

fix copy issue + add regression test (#2441)

2023-11-25 14:06:08 -08:00

test_uops.py

remove force_wait, refactor to graph (#2405)

2023-11-23 12:46:07 -08:00

test_webgpu.js

retry test_webgpu.js 3 times (#2362)

2023-11-18 21:24:47 -05:00

test_winograd.py

Profiling-helper (#2321)

2023-11-16 14:15:56 -08:00