tinygrad

Commit Graph

Author	SHA1	Message	Date
George Hotz	2d3c7e4d4e	some TestPickleJIT tests (#5860 ) * some TestPickleJIT tests * hotfix: print which opencl device we are using	2024-08-01 12:39:59 -07:00
George Hotz	9d05dfb6f4	move JIT graphing into CapturedJit (#5852 ) * move JIT graphing into CapturedJit * better * _jit_cache * clear inputs cleanup * test_pickle_jit with graph + cleanup * 0 is fine to start * support None in bufs * alloc real buffers * cleaner	2024-07-31 20:48:17 -07:00
George Hotz	c6a8395f1b	CapturedJit is fun to pickle [run_process_replay] (#5851 ) * CapturedJit is fun to pickle * export input replace	2024-07-31 17:23:01 -07:00
George Hotz	c98ca23cb9	test pickle variable (#5150 ) * test pickle variable * fix process replay	2024-06-25 19:49:21 -07:00
George Hotz	f95658bc3e	hotfix: pickle jit works if you delete the function	2024-05-05 10:14:03 -07:00
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
qazal	42edae8935	pickle schedules (#4114 ) * pickle schedules * Update test_pickle.py * Update test_pickle.py --------- Co-authored-by: George Hotz <72895+geohot@users.noreply.github.com>	2024-04-09 13:47:25 -07:00
George Hotz	ae849d12d7	numpy device + pickle it (#4120 )	2024-04-09 13:19:30 -07:00
George Hotz	086291e8c6	hotfix: add test for JIT reset	2024-03-17 21:35:49 -07:00
George Hotz	bf3e1c4df2	support pickling tensors and others (#3787 ) * test pickle tensors * pickle unrealized tensor * pickle jit, don't save Device in every CompiledASTRunner * real test of pickle, move delete	2024-03-17 18:29:14 -07:00

10 Commits