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 a71c5d59eb86290634a258704d8bab2378b8d63d.
* 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 |