George HotzandGitHub 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