George Hotz 3baaf298d6 two stage cumsum in tensor.py (#2331) ...

* two stage cumsum in tensor.py

* 2 more kernels for llama cumsum

* gpt-2 and llama use fast multinomial

2023-11-16 12:09:53 -08:00

..

test_disk_cache.py

use unique table name for each disk_cache test (#2184)

2023-10-30 13:49:49 -07:00

test_disk_tensor.py

move things, clean up extra (#2292)

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

test_flopcounter.py

reorder a few things (#1915)

2023-09-25 10:17:21 +08:00

test_helpers.py

two stage cumsum in tensor.py (#2331)

2023-11-16 12:09:53 -08:00

test_shapetracker.py

Make ShapeTracker Immutable (#1909)

2023-09-24 21:09:03 +08:00

test_shm_tensor.py

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

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

test_symbolic.py

MulNode.__lt__ rule (#2086)

2023-10-17 13:18:35 -07:00