7 Commits

Author	SHA1	Message	Date
George Hotz	8656eebb42	jit doesn't use named tensors (#2393) ... * jit doesn't use named tensors * move to compile2 * remove broken single root junk * explicit float32 * skip slow test	2023-11-23 00:13:18 -08:00
George Hotz	5aaa8a0cc1	fix shape	2023-10-31 11:36:19 -07:00
George Hotz	a27c9f9de5	openpilot compile2 (#2189) ... * try compile2 * pass to thneed * fix tanh onnx	2023-10-31 11:08:58 -07:00
George Hotz	881fd7c141	add mops to graph, refactor IMAGE (#2100) ... * add mops to graph, refactor IMAGE * no reshape pushing * add todo * fix openpilot model alt * push reshapes reduces kernels in new op * IMAGE=2 is a first class citizen now	2023-10-17 21:27:51 -07:00
George Hotz	5a4a62ecae	Disable logging in early compile2 and lower kernel counts (#2090) ... * Revert "Revert "openpilot kernel fix from 209 to 207 (#2006)" (#2065)" This reverts commit 924ecc4d6a. * gate behind OPT >= 4 * disable_logging in schedule * simple * from master * more images * revert that * 206 kernels	2023-10-16 20:15:24 -07:00
George Hotz	a7b18ac325	try beam search on device (#2085) ... * try beam search on device * fix beam with nolocals * ops too --------- Co-authored-by: Comma Device <device@comma.ai>	2023-10-16 12:52:42 -07:00
George Hotz	5472a14544	openpilot compile2 (#1977) ... * start compile2 * tweak * why are there two more kernels? * minor cleanups * don't break onnx tests * add __metadata__ support to safetensors * no early realize in onnx * cleanups * bugfix * clean up image type, add optimize * opt to match old * try that * opt work * run compile2 * optimizer * prt more * prerealize * imp * NOLOCALS works * no locals means no locals * support fractional globals * all locals welcome * int that * cleanups * show gemv regression * clean up diff * use idx for the cond * nolocals --------- Co-authored-by: Comma Device <device@comma.ai>	2023-10-15 20:39:46 -07:00