tinygrad/tinygrad
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
..
codegen NV tensor_cores in kernel.py (#4399) 2024-05-02 22:33:08 -04:00
engine subbuffer support (#4397) 2024-05-03 18:05:57 -07:00
features search: skip timing the unoptimized kernel (#4395) 2024-05-02 16:48:49 -04:00
nn update docs (#4356) 2024-04-30 16:51:42 +09:00
renderer fix define global (#4383) 2024-05-01 22:32:56 -04:00
runtime subbuffer support (#4397) 2024-05-03 18:05:57 -07:00
shape add consecutive property to shapetracker 2024-05-02 10:41:28 -07:00
__init__.py add mnist tutorial 2024-04-25 16:08:32 +08:00
buffer.py subbuffer support (#4397) 2024-05-03 18:05:57 -07:00
device.py subbuffer support (#4397) 2024-05-03 18:05:57 -07:00
dtype.py rename Scalar to ConstType and cast_scalar to as_const (#3946) 2024-03-26 22:39:58 -04:00
function.py remove downcast_half in sum (#4376) 2024-05-01 11:46:44 -04:00
helpers.py subbuffer support (#4397) 2024-05-03 18:05:57 -07:00
lazy.py subbuffer support (#4397) 2024-05-03 18:05:57 -07:00
ops.py subbuffer support (#4397) 2024-05-03 18:05:57 -07:00
tensor.py refactor sparse_categorical_crossentropy (#4406) 2024-05-03 14:28:36 -04:00