George Hotz 9fc4465557 subbuffer support (#4397) ...

* subbuffer support

* diskbuffer offset

* cuda subbuffer works

* use subbuffer

* more subbuffer tests

* consecutive

* cast

* consec

* offset

* view is a better name

* offset is in nbytes

* fix view + memory planner

* delete unused DiskRunner

* reverse order

* no subbuffers on unrealized consts

* only enabled for disk

* don't reverse memory

* view supported devices

* pickle buffer view

* ring jit

* support extra view inputs in jit

* fix JIT=2 issue

* test copy jit

* p2p isn't an option anymore

* fix dep tracking issue

* fix mypy

* fix pickle

* from_nv is contents now

2024-05-03 18:05:57 -07:00

..

__init__.py

more test cleanups (#2631)

2023-12-05 16:17:57 -08:00

test_device.py

fix UnboundLocalError when running Compiler with DISABLE_COMPILER_CACHE (#3296)

2024-02-01 21:12:33 -05:00

test_disk_cache.py

Escape table name in diskcache queries. (#3543)

2024-02-29 13:04:21 -08:00

test_disk_tensor.py

subbuffer support (#4397)

2024-05-03 18:05:57 -07:00

test_flopcounter.py

_reduce_op is axis based now (#3462)

2024-02-21 16:36:31 +01:00

test_helpers.py

hotfix: skip fetch tests always

2024-05-01 08:43:26 -07:00

test_shapetracker_math.py

remove unused expr node (#3170)

2024-01-18 14:18:43 -08:00

test_shapetracker.py

Fix in _reshape_mask (#4332)

2024-04-28 11:57:39 -04:00

test_shm_tensor.py

ops_shm removed (#2351)

2023-11-18 11:41:58 -08:00

test_symbolic.py

factor out partial in SumNode div int (#3841)

2024-03-20 16:34:33 -04:00

test_view.py

fix merging condition in merge_dims (#3363)

2024-02-12 11:50:26 +01:00