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