13 commits

Author	SHA1	Message	Date
chenyu	58d3d5030b	vars_from_ast -> LazyOp.vars (#2965)	2024-01-01 18:12:38 -05:00
George Hotz	56f44bd10e	move the compiler cache to be global (#2957) ... * move the compiler cache to be global * remove non robust test * remove dead code	2024-01-01 10:59:56 -08:00
George Hotz	00d9eda961	FROM -> COPY, move vars_from_ast (#2675)	2023-12-07 16:32:30 -08:00
chenyu	51af99367f	fix fuzz_linearizer using new device Buffer (#2674)	2023-12-07 19:21:47 -05:00
Christopher Mauri Milan	7f01dd04f0	Apply ruff linting rules to tests (#2473) ... * everything except F821 * enable F821 with noqa * dumb fix * fix remaining imports and (former) lambdas * replace _ with noqa to avoid gc	2023-11-27 21:24:06 -08:00
George Hotz	9e07824542	move device to device.py (#2466) ... * move device to device.py * pylint test --disable R,C,W,E --enable E0611 * fix tests	2023-11-27 11:34:37 -08:00
George Hotz	8e9cdef61f	clean up the buffers (#2447) ... * clean up the buffers * remove allocate_output * functools.lru_cache is methodcache * add TestShapeTrackerSize * cache_clear * no 0 sz buffer, add _ on functions that shouldn't be imported * fix size * if -> while	2023-11-26 11:02:29 -08:00
George Hotz	0505c5ea50	remove force_wait, refactor to graph (#2405) ... * remove force_wait * refactor * get rid of stupid ASTRunner * fix del in diskbuffer * BufferOps.FROM_UNDERLYING * put offset in the rawbuffer * fix bugs * use exec	2023-11-23 12:46:07 -08:00
chenyu	a98511561c	fuzz_linearizer same api for interpreted and compiled (#2320)	2023-11-15 17:40:22 -05:00
chenyu	fac8633ba8	explicit opts for test_linearizer_failures (#2299) ... * explicit opts for test_linearizer_failures * typo * update the invalid check	2023-11-14 11:52:38 -05:00
George Hotz	85d26ddc36	uops loop removal (#2262) ... * remove the loop * cleanups * tests failing still * global_loop_ctx wasn't needed * replace_op is cleaner * minor opt * cast opt was wrong * uop_num * uop num was dumb * tuplize_uops * torch tests * fix test_uops	2023-11-10 15:24:47 -08:00
chenyu	75f6e9ab54	one more fuzz linearizer failed example (#2260)	2023-11-10 09:17:37 -05:00
chenyu	10d642e174	fuzz linearizer transformation (#2188) ... * fuzz linearizer transformation * no standard normal for fp16 * work * Interpreted start * CPU and TORCH work * fix MemBuffer with same idx * id for failed kernels * no image and variable for Interpreted * symbolic shape * IMAGE only for GPU * Interpreted almost all good * cleanup * fix bufs_from_lin * zero size * some failed examples * just Exception * just test not pass	2023-11-09 08:03:27 -08:00