12 commits

Author	SHA1	Message	Date
Tobias Fischer	8c9c1cf62f	Pulled CLIP and UNet into Seperate Files (#5253) ... * pulled clip and unet into seperate files * reference cleanup, lru cache fix * better pool indexing	2024-07-01 22:33:01 -04:00
George Hotz	150ea2eb76	create engine folder and move code (#3948) ... * retry * older tf * that	2024-03-26 20:38:03 -07:00
George Hotz	41efaa848c	move graph.py and jit.py into features (#3376) ... * move graph.py into features * move jit into features * fix quickstart	2024-02-12 17:34:34 +01:00
George Hotz	a280cfe169	move dtypes to dtype.py (#2964) ... * move dtypes to dtype.py * fix urllib	2024-01-01 14:58:48 -08:00
chenyu	0723f26c80	dtypes.default_float and dtypes.default_int (#2824)	2023-12-18 12:21:44 -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	7170a9a057	coder.py can write and run code (#2439) ... * wip mistral * coder * touchups * cleanups * mistral cleanups * clean up cache create * download the weights, fix tests * fix llama loading * global fixup * clean up all * move llama model * cleanups * Revert "cleanups" This reverts commit a71c5d59eb. * fine, leave it	2023-11-25 12:27:54 -08:00
George Hotz	70a65c201e	JIT support in Interpreted (#2314) ... * factor that out * jit is supported everywhere * fix some tests * there's no jit supported device, the jit is everywhere * fix test uops	2023-11-15 11:13:38 -08:00
Roelof van Dijk	36ab04ae35	perf: lazyop as dataclass (#1603) ... * perf: lazyop as dataclass fix: linter fix: restore eq * use builtin methods, buffers to property to allow freezing * fix: reduce diff * fix: can't freeze due to KOPT tests, mypy * fix: explicit hash * can freeze if tests are fixed * fix: typo --------- Co-authored-by: Roelof van Dijk <roelof.van.dijk@vitestro.com> Co-authored-by: George Hotz <72895+geohot@users.noreply.github.com>	2023-10-25 17:54:30 -04:00
George Hotz	2d0c1037b1	Fix up latest openpilot model (#1976) ... * fix gemv triggering for gemm * fixup_openpilot * external test issues	2023-10-05 05:24:28 -07:00
nimlgen	1c0449e190	add cache collector (#1595) ... * init cache collector * add test_cache_collector.py * switch GlobalCounters.cache to CacheCollector * init jit models test * jitted SD * add debug msg to print loaded bufs count * moved cache collctor to jit * clearer SD * no double device import	2023-08-28 19:59:55 -07:00