7 Commits

Author	SHA1	Message	Date
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